Make gradient checkpointing and offloading per-component - #1476

Merged

dxqb merged 11 commits into

Nerogar:mergefrom

dxqb:split-offload

Jul 5, 2026

Commits on Jun 20, 2026

Centralize model composition and training methods in ModelType

dxqb
and
claude
committed
414ca28
Add per-component offloading/checkpointing from PR #1476
dxqb
committed
f4b7d50

Commits on Jul 1, 2026

Centralize model composition and training methods in ModelType
dxqb
committed
d680a19
Merge branch 'centralize-model-type' into split-offload
dxqb
committed
a562693

Commits on Jul 4, 2026

Centralize model composition and training methods in ModelType

dxqb
and
claude
committed
c9c88e7
Merge origin/centralize-model-type into split-offload

dxqb
and
claude
committed
338bc65
Fix Anima/Krea2 checkpointing calls to match part-based conductor API
dxqb
committed
baab41c
Merge remote-tracking branch 'Nerogar/merge' into split-offload
dxqb
committed
b5c61ce
Address PR #1476 review comments
dxqb
committed
6908a6a
Address further PR #1476 review: decouple checkpointing/offloading from train

dxqb
and
claude
committed
a30f033

Commits on Jul 5, 2026

Remove stale text_encoder None-checks in Chroma/Qwen/ZImage/Flux2 setups

dxqb
and
claude
committed
f40ba01