Bubbliiiing 1061378b66 Add bfloat16 norm && fix bug in cfg_optimization (#217)

2025-05-29 11:28:00 +08:00

..

__init__.py

Pai sparse attention (#211)

2025-05-26 16:00:58 +08:00

cfg_optimization.py

Add bfloat16 norm && fix bug in cfg_optimization (#217)

2025-05-29 11:28:00 +08:00

discrete_sampler.py

Update Wan and Wan-Fun (#125)

2025-03-26 17:49:51 +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

fp8_optimization.py

Pai sparse attention (#211)

2025-05-26 16:00:58 +08:00

lora_utils.py

Pai sparse attention (#211)

2025-05-26 16:00:58 +08:00

utils.py

Fix bug in lora training && Update zero3 cpu offload && Teacache thresholds (#209)

2025-05-13 20:54:20 +08:00