Files
mamba_diffusion/as_mamba.py
gameloader 1446f97459 refactor(as_mamba): Remove dt prediction and use fixed dt
Removes the `dt_head` network and associated configuration parameters
(dt_min, dt_max, lambda_nfe, warmup_epochs). Replaces predicted time
steps with a fixed value derived from sequence length. Eliminates
the warmup phase and NFE loss calculation.
2026-01-21 13:07:36 +08:00

15 KiB