Commit Graph
11 Commits
Author SHA1 Message Date
Bubbliiiing 1061378b66 Add bfloat16 norm && fix bug in cfg_optimization (#217) 2025-05-29 11:28:00 +08:00
Bubbliiiing a1e6ea6335 Pai sparse attention (#211)
* pai fuser sparse test

* make pai fuser more clear

* make pai fuser more clear

* make pai fuser more clear

* make pai fuser more clear

* Update Readme

* disable compile in rope for less error info

* Fix sage attention backward bug

* Fix checkpoint bugs

* Fix Attention

* Update predict and fast rope

* Fix bug in Sparse

* Fix bug in Sparse

* Fix bug in loras load

* Delete useless import

* Fix bug in ui

* Fix bug in ui
2025-05-26 16:00:58 +08:00
Bubbliiiing ed4f10beef Fix bug in lora training && Update zero3 cpu offload && Teacache thresholds (#209) 2025-05-13 20:54:20 +08:00
Bubbliiiing f26f0a809b Update cfg skip to wrapper && Update Teacache && Update Reamde (#200) 2025-05-09 18:11:53 +08:00
Bubbliiiing d11d6665bc Update sampler and print (#189) 2025-04-29 20:42:21 +08:00
Bubbliiiing 5e2f7fcdac Update Wan Fun V1.1 (#179) 2025-04-27 15:06:36 +08:00
Tianhao-Qi faa208e8d9 Add FlowDPMSolverMultistepScheduler, FlowUniPCMultistepScheduler, keep consistent with original teacache and avoid unnecessary computation (#175) 2025-04-27 14:08:13 +08:00
mohui37 3c19c391cd Update utils.py (#169) 2025-04-25 20:25:29 +08:00
Bubbliiiing f149dba611 Update Riflex && Pipeline callback bug fix && Training Code bug fix (#150) 2025-04-09 15:23:38 +08:00
Bubbliiiing ae4f4187f7 Multi GPUs API && Control preprocess nodes && Fix merge_lora bug in sequential_cpu_offload (#145) 2025-03-31 19:57:18 +08:00
Bubbliiiingandhuangkunzhe.hkz e1e7145c87 Update Wan and Wan-Fun (#125)
Update Wan and Wan-Fun

---------

Co-authored-by: huangkunzhe.hkz <huangkunzhe.hkz@alibaba-inc.com>
2025-03-26 17:49:51 +08:00