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

Refactor qwen5b model code to be qwen 5b specific

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-06 10:54:56 -06:00
parent 5d8922fca2
commit 14ccf2f3ce
4 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/diffusion_models/wan22/__init__.py
View File

@@ -1 +1 @@
from .wan22_model import Wan22Model
from .wan22_5b_model import Wan225bModel

2
extensions_built_in/diffusion_models/wan22/wan22_model.py → extensions_built_in/diffusion_models/wan22/wan22_5b_model.py
View File

@@ -80,7 +80,7 @@ def time_text_monkeypatch(
return temb, timestep_proj, encoder_hidden_states, encoder_hidden_states_image
class Wan22Model(Wan21):
class Wan225bModel(Wan21):
arch = "wan22_5b"
_wan_generation_scheduler_config = scheduler_configUniPC
_wan_expand_timesteps = True