bubbliiiing 03ef7ec62b Update self forcing code

2026-10-23 13:20:38 +08:00

..

__init__.py

Update self forcing code

2026-10-23 13:20:38 +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

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 and reward model training

2026-10-12 22:25:49 +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 self forcing code

2026-10-23 13:20:38 +08:00

fsdp_utils.py

Update self forcing code

2026-10-23 13:20:38 +08:00

gpu_memory_mode.py

Upgrade GPU Memory Mode architecture (#521)

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

group_offload.py

Upgrade GPU Memory Mode architecture (#521)

2026-09-29 19:18:17 +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