AstrAI/astrai/model
ViperEkura 9096e413c3 refactor: RotaryEmbedding 合并 cos/sin 为单一复数缓存
- get_rotary_emb() 返回复数张量替代 Tuple[cos, sin]
- RotaryEmbedding 存储单一 freqs_cis buffer 替代分离的 cos_cached/sin_cached
- forward 中 view_as_complex 重建复数
2026-05-15 18:03:59 +08:00
..
__init__.py chore: 修改类名,优化导入顺序 2026-04-05 22:27:57 +08:00
automodel.py refactor: 消除多处重复模式,统一工厂和参数传递 2026-05-14 18:00:50 +08:00
module.py refactor: RotaryEmbedding 合并 cos/sin 为单一复数缓存 2026-05-15 18:03:59 +08:00
transformer.py refactor: 重构 cache 和 inference 参数体系,分离存储与分配 2026-05-14 20:05:08 +08:00