refactor(project): 重构项目文件组织结构
- 修改模型管理和训练页面的导入路径 - 更新 main.py 中的导入模块 - 调整 tools 包的内容,移除 model 模块 - 新建 train 包,包含 model 模块 - 优化 __init__.py 文件,简化导入语句
This commit is contained in:
parent
5a21c8598a
commit
868fcd45ba
@ -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是当前工作目录下的一个文件夹
|
||||
|
@ -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:
|
||||
|
2
main.py
2
main.py
@ -1,5 +1,5 @@
|
||||
import gradio as gr
|
||||
import unsloth
|
||||
import train
|
||||
from frontend import *
|
||||
from db import initialize_sqlite_db, initialize_prompt_store
|
||||
from global_var import init_global_var, get_sql_engine, get_prompt_store
|
||||
|
@ -1,6 +1,5 @@
|
||||
from .parse_markdown import parse_markdown
|
||||
from .parse_markdown import *
|
||||
from .document import *
|
||||
from .json_example import generate_example_json
|
||||
from .model import *
|
||||
from .port import *
|
||||
from .reasoning import call_openai_api
|
1
train/__init__.py
Normal file
1
train/__init__.py
Normal file
@ -0,0 +1 @@
|
||||
from .model import *
|
Loading…
x
Reference in New Issue
Block a user