carry/gzhu-biyesheji
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
91 Commits 4 Branches 0 Tags
Commit Graph

46 Commits

Author SHA1 Message Date
carry
4f7926aec6 feat(train_page): 实现训练目录自动递增功能 
- 在 training 文件夹下创建递增的目录结构
- 确保 training 文件夹存在
- 扫描现有目录，生成下一个可用的目录编号
- 更新训练模型函数，使用新的训练目录
 2025-04-14 16:46:29 +08:00
carry
11a3039775 fix(train_page): 修正模型训练保存路径 2025-04-14 16:31:00 +08:00
carry
088067d335 train: 更新模型训练功能和日志记录方式 
- 修改训练目录结构，将检查点和日志分开保存
- 添加 TensorBoard 日志记录支持
- 移除自定义 LossCallback 类，简化训练流程
- 更新训练参数和回调机制，提高代码可读性
- 在 requirements.txt 中添加 tensorboardX 依赖
 2025-04-14 16:19:37 +08:00
carry
9fb31c46c8 feat(train): 添加训练过程中的日志记录和 loss 可视化功能 
- 新增 LossCallback 类，用于在训练过程中记录 loss 数据
- 在训练模型函数中添加回调，实现日志记录和 loss 可视化
- 优化训练过程中的输出信息，增加当前步数和 loss 值的打印
 2025-04-14 15:18:14 +08:00
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
bb1d8fbd38 feat(train_page): 添加训练 Loss 曲线显示功能 
- 在训练页面添加了 Loss 曲线图表
- 实现了 GradioLossCallback 类用于记录训练过程中的 Loss 数据
- 修改了训练函数，通过回调函数收集 Loss 信息并更新图表
- 优化了训练函数的返回值结构，支持同时返回文本日志和 Loss 数据
 2025-04-13 21:49:43 +08:00
carry
4558929c52 fix: 调整了import的顺序，让unsloth最先import以提高性能 2025-04-13 21:35:47 +08:00
carry
0722748997 feat(train_page): 添加 LoRA 秩动态输入功能 
- 在训练页面新增 LoRA 秩输入框，使用户可以动态设置 LoRA 秩
- 更新训练模型函数，添加 LoRA 秩参数并将其用于模型配置
- 保留原有功能，仅增加 LoRA 秩相关配置
 2025-04-13 21:12:02 +08:00
carry
e08f0059bb feat(train_page): 优化训练过程以专注于响应生成 
- 引入 train_on_responses_only 函数，用于优化训练过程
- 设置 instruction_part 和 response_part 参数，以适应特定的对话格式
- 此修改旨在提高模型在生成响应方面的性能和效率
 2025-04-13 21:05:14 +08:00
carry
79d3eb153c refactor(train_page): 优化训练页面布局和功能 
- 移除了 max_steps_input 组件，减少不必要的输入项
- 将 per_device_train_batch_size_input 和 epoch_input 的标签简化为 "batch size" 和 "epoch"
- 新增 save_steps_input 组件，用于设置保存步数
- 修改 train_model 函数，移除了 max_steps 参数
- 更新了 trainer.train() 方法的调用，设置 resume_from_checkpoint=False
 2025-04-13 01:56:10 +08:00
carry
2d39b91764 feat(train_page): 添加模型训练超参数配置功能 
- 新增学习率、批次大小、最大训练步数等超参数输入组件
- 实现超参数在训练过程中的动态应用
- 调整训练参数以适应不同硬件环境
- 优化训练过程，支持按步数保存模型
 2025-04-13 01:04:27 +08:00
carry
539e14d39c feat(frontend): 完成了前端微调的代码逻辑 2025-04-12 18:42:22 +08:00
carry
611904cef9 feat(frontend): 添加数据集选择功能到训练页面 
- 在 train_page.py 中添加数据集选择下拉框
- 从全局变量中获取数据集列表并设置初始值
- 添加交互性和自定义值支持
 2025-04-11 19:43:34 +08:00
carry
a23ad88769 fix(frontend): 修复删除提示功能中的数据库连接错误 
- 将 prompt_store 更改为 get_prompt_store()，以解决数据库连接未建立的问题
- 优化了删除提示功能的代码，提高了系统稳定性
 2025-04-11 18:53:17 +08:00
carry
83427aaaba feat(frontend): 增加超参数设置并优化聊天页面布局 
- 在聊天页面添加了超参数输入框，包括最大生成长度、温度、Top-p 采样和重复惩罚
- 优化了聊天框的布局，使用 gr.Row() 和 gr.Column() 实现了更合理的界面结构
- 更新了 bot 函数，支持根据用户输入的超参数进行文本生成
- 修复了一些代码格式问题，提高了代码的可读性
 2025-04-11 18:48:13 +08:00
carry
61672021ef fix(frontend): 修复聊天页面并的流式回复 
- 导入 Thread 和 TextIteratorStreamer 以支持流式生成
- 重新设计 user 和 bot 函数，优化对话历史处理
- 添加异常处理和错误信息显示
- 改进模型和分词器的加载逻辑
- 优化聊天页面布局和交互
 2025-04-11 18:33:31 +08:00
carry
fb6157af05 feat(frontend): 初步实现聊天页面的智能回复功能 2025-04-11 18:08:38 +08:00
carry
ab7897351a fix(global_var): 修复全局变量多文件多副本的不统一问题 2025-04-11 18:04:42 +08:00
carry
0fa2b51a79 refactor(frontend): 优化模型管理页面的交互和显示 
- 将状态输出从 Textbox 改为 Label 组件，提高用户体验
- 添加 get_model_name 函数以获取模型名称，提高代码复用性
- 更新模型加载、卸载和保存后的状态显示，使信息更加准确
- 优化模型列表刷新功能，确保模型列表实时更新
 2025-04-11 00:14:40 +08:00
carry
2e552c186d refactor(frontend): 重构模型选择界面的变量命名 
- 将模型选择的 Dropdown 组件从 dropdown 重命名为 model_select_dropdown，提高代码可读性
- 更新 load_button 和 refresh_button 的输出目标，以适应新的变量名
 2025-04-10 21:19:58 +08:00
carry
1b3f546669 refactor(frontend): 重构前端页面并添加独立运行功能 
- 在 chat_page 和 prompt_manage_page 中添加了独立运行的入口
- 引入 sys 和 pathlib 模块以支持路径操作
- 修改了模块导入方式，使其能够作为独立脚本运行
- 优化了代码结构，提高了可读性和可维护性
 2025-04-10 21:18:05 +08:00
carry
402bc73dce feat(model_manage_page): 增加模型保存和刷新功能 
- 新增保存模型功能，用户可以输入模型名称并保存当前加载的模型
- 添加刷新模型列表按钮，用户可以随时更新模型下拉菜单中的选项
- 优化页面布局，使按钮和输入框更加合理地排列
 2025-04-10 20:18:03 +08:00
carry
a407fa1f76 feat(model_manage_page): 实现模型加载和卸载功能 
- 添加模型加载和卸载按钮
- 实现模型加载和卸载的逻辑
- 添加相关模块的导入
- 扫描模型目录并显示在下拉框中
 2025-04-10 19:52:08 +08:00
carry
e7cc03297b feat(frontend): 添加了简单聊天机器人页面 2025-04-10 17:38:02 +08:00
carry
051d1a7535 feat(frontend): 添加模型管理页面并初始化模型相关全局变量 
- 在 frontend/__init__.py 中添加 model_manage_page 模块引用
- 新增 model_manage_page.py 文件，实现模型管理页面的基本框架
- 在 global_var.py 中添加 model 和 tokenizer 全局变量
- 在 main.py 中集成模型管理页面到主应用的 Tabs 组件中
 2025-04-10 17:37:45 +08:00
carry
97172f9596 feat(dataset): 设置问答数据集展示页面的每页显示数量 
- 在 dataset_manage_page 函数中添加 samples_per_page 参数
- 设置每页显示的样本数量为 20 条
 2025-04-10 16:12:59 +08:00
carry
8fb9f785b9 feat(frontend): 展示数据集管理页面的问答数据 
- 添加 QA 数据集展示组件
- 实现数据集选择时动态加载对应的问答数据
- 优化数据集管理页面布局
 2025-04-09 22:23:55 +08:00
carry
2c8e54bb1e feat(dataset): 初步完成数据集管理页面和功能 2025-04-09 20:49:20 +08:00
carry
4d8754aad2 feat(frontend): 实现数据集生成页面的文档和模板选择功能 
- 添加文档和模板的下拉选择框
- 实现文档和模板选择后的状态更新
- 优化页面布局，分为文档和模板两个列
 2025-04-09 17:19:40 +08:00
carry
6a00699472 feat(frontend): 实现提示词模板管理页面 
- 添加获取、添加、编辑和删除提示词功能
- 实现数据表格展示和操作
 2025-04-09 11:08:18 +08:00
carry
9741ce6b92 refactor(db): 优化了代码，调整了import顺序，删除了无用变量 2025-04-09 10:19:57 +08:00
carry
374b124cf8 feat(setting_page): 添加供应商后清空输入框 
- 修改 add_provider 函数，返回清空后的输入框值
- 更新 add_button.click 事件处理，添加清空输入框的输出
 2025-04-09 08:17:43 +08:00
carry
74ae5e1426 refactor(db): 重命名数据库引擎获取函数 
将 get_engine 函数重命名为 get_sqlite_engine，以更清晰地表示其功能和用途。
- 更新了 db/__init__.py 中的导入和 __all__ 列表
- 修改了 db/init_db.py 中的函数定义
- 更新了前端设置页面和全局变量中的导入和函数调用

此更改提高了代码的可读性和维护性，特别是在将来可能添加其他类型数据库引擎的情况下。
 2025-04-09 08:12:59 +08:00
carry
0a6ae7a4ee feat(frontend): 重构前端页面并添加新功能 
- 重命名 dataset_page 为 prompt_manage_page，支持提示词模板管理
- 新增 dataset_generate_page 和 dataset_manage_page 页面
- 更新 main.py 中的页面引用和标签名称
- 修改前端初始化文件，使用 * 导入所有页面模块
 2025-04-09 08:11:40 +08:00
carry
faf72d1e99 feat(frontend): 完成了编辑 API Provider 功能 2025-04-09 08:04:40 +08:00
carry
cce5e4e114 feat(frontend): 完成了 API Provider 删除和添加了编辑功能的函数 2025-04-09 00:48:22 +08:00
carry
293f63017f feat(frontend): 添加 API Provider 表格选中行状态监听 
- 新增选中行的全局变量 selected_row
- 实现 select_record 函数来保存选中行数据
- 在表格中添加选中行事件监听
- 优化代码结构，提高可读性和可维护性
 2025-04-09 00:37:15 +08:00
carry
dc28c25c65 feat(frontend): 更新设置页面按钮样式 
- 为"添加新API"按钮添加 primary 样式
- 为"编辑选中行"按钮添加 primary 样式
- 为"删除选中行"按钮添加 stop 样式
- 保持"刷新数据"按钮的 secondary 样式
 2025-04-08 14:23:31 +08:00
carry
46b4453ccd refactor(frontend): 重构数据库连接方式 
- 移除各前端页面中重复的数据库引擎初始化代码
- 在 global_var.py 中统一初始化和存储数据库引擎
- 更新 setting_page.py 和 main.py 中的数据库连接逻辑
- 优化代码结构，提高可维护性和可扩展性
 2025-04-08 13:19:58 +08:00
carry
286db405ca feat(frontend): 优化设置页面并添加数据刷新功能 
- 为 get_providers 函数添加异常处理，提高数据获取的稳定性
- 在设置页面添加刷新按钮，用户可手动触发数据刷新
- 优化页面布局，调整组件间距和对齐方式
 2025-04-07 23:17:43 +08:00
carry
d40f5b1f24 fix(frontend): 优化 API Provider 添加功能并处理异常 
- 为 model_id、base_url 和 api_key 添加空值检查，避免无效输入
- 添加异常处理，确保在出现错误时能够及时响应并提示用户
- 优化 add_provider 函数，提高代码可读性和健壮性
 2025-04-07 13:02:45 +08:00
carry
7a77f61ee6 feat(frontend): 添加 API Provider 的增加功能 2025-04-07 00:28:52 +08:00
carry
841e14a093 feat(frontend): 添加数据集页面并重构主页面布局 
- 新增 dataset_page 模块，实现数据集页面的基本布局
- 重构 main.py 中的页面加载方式，使用列表收集所有页面
- 更新主页面布局，将聊天页面作为第一个选项卡
- 调整设置页面的加载方式，直接使用函数调用
 2025-04-06 22:49:37 +08:00
carry
2ff077bb1c refactor(frontend): 重构前端页面导入方式 
- 在 main.py 中使用更简洁的导入方式
- 新增 __init__.py 文件以简化前端页面的导入
 2025-04-06 22:46:31 +08:00
carry
513b639bce feat(frontend): 添加了设置页面的api provider展示 2025-04-06 22:05:56 +08:00
carry
519a5f3773 feat(frontend): 添加前端页面模块并实现基本布局 
- 新增 chat_page.py、setting_page.py 和 train_page.py 文件，分别实现聊天、设置和微调页面的基本布局
- 添加 main.py 文件，集成所有页面并创建主应用
- 在 requirements.txt 中添加 gradio 依赖
 2025-04-06 14:49:01 +08:00