auto-round
Enable NextStepDiffusion and support multi-device tuning for diffusion
#1640
Merged
Merged
Enable NextStepDiffusion and support multi-device tuning for diffusion #1640
xin3he
merged
24 commits into
main
from
xinhe/3-30a
fix nextstep loading issue
2bc36971
support 6.0.0 gptqmodel
2bb744ac
wenhuach21
commented
on 2026-03-31
Enhance DiffusionCompressor with custom pipeline support and improve …
576e1308
Merge branch 'main' into xinhe/3-30a
010d4c0e
fix bug for nextstep
bb22b290
set self.num_inference_steps=1 for calibration
87c40378
Remove unused function; add CUDA CI for diffusion tuning test; revert…
1c012007
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&s=60)
[pre-commit.ci] auto fixes from pre-commit.com hooks
240df28f
fix bug
04bb9100
Refactor device dispatching logic; remove unused function and update …
d84c2d24
[pre-commit.ci] auto fixes from pre-commit.com hooks
86328025
xin3he
changed the title 
fix z-image save_pretrained
e2926de8
[pre-commit.ci] auto fixes from pre-commit.com hooks
e9967c92
fix pylint
816d2e29
revert "force num_inference_steps=1"
6a4fdc84
[pre-commit.ci] auto fixes from pre-commit.com hooks
c9597d9f
fix pylint
55eb544c
Merge branch 'main' into xinhe/3-30a
b2f40a27
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
wenhuach21
commented
on 2026-04-10
support diffuser loading
9a716a2a
[pre-commit.ci] auto fixes from pre-commit.com hooks
3301eb1f
mengniwang95
commented
on 2026-04-13
mini update, mv loading to special model handler
81513a8d
update try except to make code simple
84bfb62a
add multi-card test
79dcdf4d
Merge branch 'main' into xinhe/3-30a
3b94c0ca
mengniwang95
approved these changes
on 2026-04-14
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub