bubbliiiing e4c74a12f2 Update phantom && add fsdp training && Fix bug in image encoder while using sage attention

2025-06-09 08:25:49 +00: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

Update phantom && add fsdp training && Fix bug in image encoder while using sage attention

2025-06-09 08:25:49 +00:00