olivame 7338cc384a ddim.py — torch.float16 → torch.bfloat16，修复 dtype 不匹配 ...

attention.py — 4 处 softmax 都包裹了 torch.amp.autocast('cuda', enabled=False)，阻止 autocast 将 bf16 提升到 fp32

2026-02-08 17:02:05 +00:00

40 KiB

Raw Blame History

View Raw