AstrAI

Files

T

ViperEkura 133a9de98f feat: _generate_streaming 支持 batch 模式

- _Result.append 存储 (idx, token) 元组，pop_all 返回对应列表
- 单 prompt: Generator[str]（向后兼容）
- 多 prompt: Generator[Tuple[int, str]]，token 交错到达，调用方自行分流
- 不使用 dispatch 线程 / Queue，避免同步开销和内存积压

2026-05-10 17:42:20 +08:00

config

refactor: 移除 device_ids 参数设计，统一通过 CUDA_VISIBLE_DEVICES 控制 GPU 分配；更新 README 训练示例

2026-05-09 14:55:43 +08:00

dataset

chore: 修改错误拼写

2026-04-06 10:37:19 +08:00

inference

feat: _generate_streaming 支持 batch 模式

2026-05-10 17:42:20 +08:00

model

fix: 修复 CLI 参数缺失/重复、device_ids 越界、generate 参数名不一致、scheduler 时序、非流式截断等 bug