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
toolkit/models/wan21/wan21.py
View File

@@ -491,7 +491,9 @@ class Wan21(BaseModel):
self.tokenizer = tokenizer
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()
if self.model_config.low_vram:
pipeline = AggressiveWanUnloadPipeline(
vae=self.vae,

4
toolkit/models/wan21/wan21_i2v.py
View File

@@ -369,7 +369,9 @@ class Wan21I2V(Wan21):
def get_generation_pipeline(self):
scheduler = UniPCMultistepScheduler(**scheduler_configUniPC)
# todo unipc got broken in a diffusers update. Use euler for now.
# scheduler = UniPCMultistepScheduler(**self._wan_generation_scheduler_config)
scheduler = self.get_train_scheduler()
if self.model_config.low_vram:
pipeline = AggressiveWanI2VUnloadPipeline(
vae=self.vae,