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))
|
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是当前工作目录下的一个文件夹
|
||||||
|
@ -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:
|
||||||
|
2
main.py
2
main.py
@ -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
|
||||||
|
@ -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
1
train/__init__.py
Normal file
@ -0,0 +1 @@
|
|||||||
|
from .model import *
|
Loading…
x
Reference in New Issue
Block a user