Commit Graph
25 Commits
Author SHA1 Message Date
bubbliiiing 499eef12c3 Update Wan2.2 UI 2025-07-30 10:48:59 +08:00
Bubbliiiing 9c05f3a741 Fix bug in Wan2.2 (#265) 2025-07-29 20:22:28 +08:00
Bubbliiiing 2d3a5cc7d6 Update Wan2.2 (#264) 2025-07-29 00:06:55 +08:00
Bubbliiiing a17b35acfb Update Torch Compile Node in Comfyui (#253) 2025-07-21 15:22:13 +08:00
huangkunzhe.hkz 9c16b3a73f Set tensor_layout in sageattn 2025-06-27 17:58:41 +08:00
Bubbliiiing 097208e817 Add checkpointing in head && Move gather to the back of the head (#231) 2025-06-25 15:21:53 +08:00
Bubbliiiing 86028a9ce6 Update fix sample size && Fix vram memory bug in wan2.1 lora training (#227) 2025-06-24 10:09:23 +08:00
bubbliiiing 95e024c681 Update Transformer3d 2025-06-10 05:02:41 +00:00
bubbliiiing a4476549e4 Update Training Code 2025-06-10 03:54:11 +00:00
bubbliiiing e4c74a12f2 Update phantom && add fsdp training && Fix bug in image encoder while using sage attention 2025-06-09 08:25:49 +00:00
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 7da0e755eb Compile && The model_full_load_and_qfloat8 mode && Move teacache to ui and api. (#188)
* Update Compile

* Update sage

* Update comment

* Update Teacache in ui && Update torch compile

* Update Ref Image in api

* Update ui

* Update ui

* Update ui

* fix bug in ui

* Update ui

* fix bug in ui

* Update longer length in ui

* Fix bug in api check

* Fix bug in pipeline

* Update model_full_load_and_qfloat8

* Update post_to_host

* Update post_to_host
2025-04-29 15:47:30 +08:00
Bubbliiiing 5e2f7fcdac Update Wan Fun V1.1 (#179) 2025-04-27 15:06:36 +08:00
hkz 1fb0696fdc Fix low_cpu_mem_usage with diffusers>=0.33.0 (#181) 2025-04-27 14:57:55 +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
Tianhao-Qi 278a4c3cd2 fix bugs in if_else conditions for function get_teacache_coefficients (#173) 2025-04-17 19:06:37 +08:00
Bubbliiiing 991e7cd2bd Api infer support (#167) 2025-04-14 18:54:13 +08:00
hkz 652b58f24c fix teacache (#160) 2025-04-09 15:24:08 +08:00
Bubbliiiing f149dba611 Update Riflex && Pipeline callback bug fix && Training Code bug fix (#150) 2025-04-09 15:23:38 +08:00
hkzandbubbliiiing 09e7bd2bd0 Support low_cpu_mem_usage=True for the text encoder of Wan2.1 (#146)
---------

Co-authored-by: bubbliiiing <3323290568@qq.com>
2025-04-01 11:35:42 +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