carry
|
1a2ca3e244
|
refactor(train): 重构训练功能并移至新模块
- 将训练逻辑从 train_page.py 移至 tools/model.py
- 新增 train_model 函数,包含完整的训练流程
- 更新 train_page.py 中的回调函数,使用新的训练函数
- 移除了 train_page.py 中未使用的导入
|
2025-04-14 14:17:04 +08:00 |
|
carry
|
9784f2aed3
|
fix(tools): 修正__init__.py使得model.py正确导入
|
2025-04-12 01:16:07 +08:00 |
|
carry
|
8a9a080745
|
refactor(tools): 移除未使用的导入语句
移除了 tools/model.py 文件中未使用的 get_chat_template 导入语句。这个修改提高了代码的可读性和维护性。
|
2025-04-11 19:43:19 +08:00 |
|
carry
|
216bfe39ae
|
feat(tools): 添加格式化对话数据的函数
- 新增 formatting_prompts_func 函数,用于格式化对话数据
- 该函数将问题和答案组合成对话形式,并使用 tokenizer.apply_chat_template 进行格式化
- 更新 imports,添加了 unsloth.chat_templates 模块
|
2025-04-11 17:56:46 +08:00 |
|
carry
|
cbb3a09dd8
|
feat(tools): 添加模型名称获取函数
- 在 tools 目录下新增 model.py 文件
- 实现 get_model_name 函数,用于获取模型的名称
- 更新 tools/__init__.py,导入新的 get_model_name 函数
|
2025-04-10 22:05:04 +08:00 |
|