ViperEkura c95ace41aa fix: prefill 时 attention mask 长度不足导致 expand 崩溃 ... - astrai/inference/scheduler.py: prefill input_mask 由 [batch, seq_len] 改为 [batch, prompt_len]，覆盖全部 KV 位置 - astrai/model/transformer.py: seq_mask is None 分支补全 start_pos + seq_len 列，避免 expand 非 singleton 维度不匹配		2026-05-10 19:56:41 +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	fix: prefill 时 attention mask 长度不足导致 expand 崩溃	2026-05-10 19:56:41 +08:00
model	fix: prefill 时 attention mask 长度不足导致 expand 崩溃	2026-05-10 19:56:41 +08:00
parallel	refactor: 移除 device_ids 参数设计，统一通过 CUDA_VISIBLE_DEVICES 控制 GPU 分配；更新 README 训练示例	2026-05-09 14:55:43 +08:00
tokenize	fix: 修复 CLI 参数缺失/重复、device_ids 越界、generate 参数名不一致、scheduler 时序、非流式截断等 bug	2026-05-09 14:36:42 +08:00
trainer	feat: Checkpoint 支持 extra 通用扩展数据，用户通过函数自定义保存/恢复优化器等状态	2026-05-09 15:50:38 +08:00
__init__.py	release: v1.3.4	2026-05-10 15:59:18 +08:00
factory.py	chore: 修改类名，优化导入顺序	2026-04-05 22:27:57 +08:00
serialization.py	feat: Checkpoint 支持 extra 通用扩展数据，用户通过函数自定义保存/恢复优化器等状态	2026-05-09 15:50:38 +08:00