mamba_diffusion

Go to file

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

.gitignore

feat(mamba): add Mamba2 implementation

2026-01-21 12:54:49 +08:00

.python-version

feat(mamba): add Mamba2 implementation

2026-01-21 12:54:49 +08:00

as_mamba.py

refactor(as_mamba): Remove dt prediction and use fixed dt

2026-01-21 13:07:36 +08:00

main.py

refactor(as_mamba): Remove dt prediction and use fixed dt

2026-01-21 13:07:36 +08:00

mamba2_minimal.py

feat(mamba): add Mamba2 implementation

2026-01-21 12:54:49 +08:00

pyproject.toml

feat(mamba): add Mamba2 implementation

2026-01-21 12:54:49 +08:00

README.md

feat(mamba): add Mamba2 implementation

2026-01-21 12:54:49 +08:00

uv.lock

feat(mamba): add Mamba2 implementation

2026-01-21 12:54:49 +08:00

README.md

The file is empty.