feat: 更新 README,添加生成长度控制和测试指令;修改 main_chat.py,优化模型加载和生成逻辑;新增单元测试文件 test_smoke.py

This commit is contained in:
2026-07-08 11:48:19 +08:00
parent 1de4656eb8
commit 24b56ebb6b
5 changed files with 177 additions and 20 deletions

View File

@@ -123,6 +123,10 @@ python main_chat.py --mode chat
# 一次性生成
python main_chat.py --mode generate --prompt "To be"
# 控制生成长度;默认使用快速 Euler 推理,--accurate 改用 RK4
python main_chat.py --mode generate --prompt "学而时习之" --n-new 40
python main_chat.py --mode generate --prompt "学而时习之" --accurate
```
对话特性:
@@ -140,6 +144,12 @@ AI: ...
你: /quit
```
## 测试
```bash
python -m unittest discover -s tests -v
```
## 外挂模块系统(可热插拔)
```python