refactor(project): 重构项目文件组织结构

- 修改模型管理和训练页面的导入路径
- 更新 main.py 中的导入模块
- 调整 tools 包的内容,移除 model 模块
- 新建 train 包,包含 model 模块
- 优化 __init__.py 文件,简化导入语句
This commit is contained in:
carry 2025-04-19 21:49:19 +08:00
parent 5a21c8598a
commit 868fcd45ba
6 changed files with 6 additions and 5 deletions

View File

@ -7,7 +7,7 @@ import torch
sys.path.append(str(Path(__file__).resolve().parent.parent)) sys.path.append(str(Path(__file__).resolve().parent.parent))
from global_var import get_model, get_tokenizer, set_model, set_tokenizer from global_var import get_model, get_tokenizer, set_model, set_tokenizer
from tools.model import get_model_name from train import get_model_name
def model_manage_page(): def model_manage_page():
workdir = "workdir" # 假设workdir是当前工作目录下的一个文件夹 workdir = "workdir" # 假设workdir是当前工作目录下的一个文件夹

View File

@ -8,7 +8,8 @@ from transformers import TrainerCallback
sys.path.append(str(Path(__file__).resolve().parent.parent)) sys.path.append(str(Path(__file__).resolve().parent.parent))
from global_var import get_model, get_tokenizer, get_datasets, get_workdir from global_var import get_model, get_tokenizer, get_datasets, get_workdir
from tools import train_model, find_available_port from tools import find_available_port
from train import train_model
def train_page(): def train_page():
with gr.Blocks() as demo: with gr.Blocks() as demo:

View File

@ -1,5 +1,5 @@
import gradio as gr import gradio as gr
import unsloth import train
from frontend import * from frontend import *
from db import initialize_sqlite_db, initialize_prompt_store from db import initialize_sqlite_db, initialize_prompt_store
from global_var import init_global_var, get_sql_engine, get_prompt_store from global_var import init_global_var, get_sql_engine, get_prompt_store

View File

@ -1,6 +1,5 @@
from .parse_markdown import parse_markdown from .parse_markdown import *
from .document import * from .document import *
from .json_example import generate_example_json from .json_example import generate_example_json
from .model import *
from .port import * from .port import *
from .reasoning import call_openai_api from .reasoning import call_openai_api

1
train/__init__.py Normal file
View File

@ -0,0 +1 @@
from .model import *