github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Hotfix some issues with Wan models caused by diffusers and transformers updates

Browse Source
This commit is contained in:
Jaret Burkett
2026-04-16 20:53:50 +00:00
parent 0c18b39346
commit ab1ee4df34
5 changed files with 36 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions_built_in/diffusion_models/wan22/wan22_14b_model.py
View File

@@ -375,7 +375,9 @@ class Wan2214bModel(Wan21):
return transformer
def get_generation_pipeline(self):
scheduler = UniPCMultistepScheduler(**self._wan_generation_scheduler_config)
# todo unipc got broken in a diffusers update. Use euler for now.
# scheduler = UniPCMultistepScheduler(**self._wan_generation_scheduler_config)
scheduler = self.get_train_scheduler()
pipeline = Wan22Pipeline(
vae=self.vae,
transformer=self.model.transformer_1,

4
extensions_built_in/diffusion_models/wan22/wan22_5b_model.py
View File

@@ -116,7 +116,9 @@ class Wan225bModel(Wan21):
return 32
def get_generation_pipeline(self):
scheduler = UniPCMultistepScheduler(**self._wan_generation_scheduler_config)
# todo unipc got broken in a diffusers update. Use euler for now.
# scheduler = UniPCMultistepScheduler(**self._wan_generation_scheduler_config)
scheduler = self.get_train_scheduler()
pipeline = Wan22Pipeline(
vae=self.vae,
transformer=self.model,