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

..

world_model_interaction_prompts

init

2026-02-08 03:29:15 +00:00

output.log

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

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

psnr_result1.json

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

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

psnr_result.json

所有case的baseline，amd版本的ground truth都上传了

2026-02-08 09:42:14 +00:00

run_world_model_interaction.sh

全链路 bf16 混合精度修正与 UNet FLOPS profiling

2026-02-08 16:01:30 +00:00