bubbliiiing d62776f113 Update model_full_load mode

2026-11-04 14:42:59 +08:00

..

__init__.py

Update Sparkwan, Reward Model Training, FastVideo (#525)

2026-10-28 11:34:50 +08:00

cfg_optimization.py

Update Z Image Control Distill Training Code and Update LogValidation in QwenImage and ZImage (#435)

2026-01-16 15:05:49 +08:00

checkpoint_utils.py

Update text_encoder_offload and accelerate saving hook

2026-11-03 17:37:27 +08:00

discrete_sampler.py

Support Qwen-Image2.1 && Support new Control for Minimax-H3 && Support Wan2.2 with 2.2VAE && Update new tqdm for All Training Code (#511)

2026-09-22 20:41:39 +08:00

flex_chunking.py

Update Sparkwan, Reward Model Training, FastVideo (#525)

2026-10-28 11:34:50 +08:00

fm_solvers_minimax_h3.py

Support Qwen-Image2.1 && Support new Control for Minimax-H3 && Support Wan2.2 with 2.2VAE && Update new tqdm for All Training Code (#511)

2026-09-22 20:41:39 +08:00

fm_solvers_unipc.py

Add FlowDPMSolverMultistepScheduler, FlowUniPCMultistepScheduler, keep consistent with original teacache and avoid unnecessary computation (#175)

2025-04-27 14:08:13 +08:00

fm_solvers.py

Add FlowDPMSolverMultistepScheduler, FlowUniPCMultistepScheduler, keep consistent with original teacache and avoid unnecessary computation (#175)

2025-04-27 14:08:13 +08:00

fp4_optimization.py

Upgrade GPU Memory Mode architecture (#521)

2026-09-29 19:18:17 +08:00

fp8_optimization.py

Upgrade GPU Memory Mode architecture (#521)

2026-09-29 19:18:17 +08:00

fsdp_ema.py

Update ema and update taehv

2026-10-29 10:59:15 +08:00

fsdp_utils.py

Update text_encoder_offload and accelerate saving hook

2026-11-03 17:37:27 +08:00

gpu_memory_mode.py

Update model_full_load mode

2026-11-04 14:42:59 +08:00

group_offload.py

fix(offload): resolve the group offloading transfer stream from the onload device (#528)

2026-10-27 23:15:38 +08:00

lora_utils_pdd.py

Add PDD for MiniMax-H3 (#515)

2026-09-04 17:02:21 +08:00

lora_utils.py

Upgrade GPU Memory Mode architecture (#521)

2026-09-29 19:18:17 +08:00

perf_metrics.py

Support Qwen-Image2.1 && Support new Control for Minimax-H3 && Support Wan2.2 with 2.2VAE && Update new tqdm for All Training Code (#511)

2026-09-22 20:41:39 +08:00

sd3_sde_with_logprob.py

Support Qwen-Image2.1 && Support new Control for Minimax-H3 && Support Wan2.2 with 2.2VAE && Update new tqdm for All Training Code (#511)

2026-09-22 20:41:39 +08:00

tqdm_bar.py

Support Qwen-Image2.1 && Support new Control for Minimax-H3 && Support Wan2.2 with 2.2VAE && Update new tqdm for All Training Code (#511)

2026-09-22 20:41:39 +08:00

trigflow_sampler.py

Support Qwen-Image2.1 && Support new Control for Minimax-H3 && Support Wan2.2 with 2.2VAE && Update new tqdm for All Training Code (#511)

2026-09-22 20:41:39 +08:00

utils_yolo.py

Update Qwen Image 2.1 Control and Flex Forcing (#519)

2026-09-24 15:52:16 +08:00

utils.py

Support Qwen-Image2.1 && Support new Control for Minimax-H3 && Support Wan2.2 with 2.2VAE && Update new tqdm for All Training Code (#511)

2026-09-22 20:41:39 +08:00