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))
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():
workdir = "workdir" # 假设workdir是当前工作目录下的一个文件夹

View File

@@ -8,7 +8,8 @@ from transformers import TrainerCallback
sys.path.append(str(Path(__file__).resolve().parent.parent))
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():
with gr.Blocks() as demo: