Skip to content

videogen
videogen
Commit 4a5213f8 authored by Stefy Lanza (nextime / spora )'s avatar Stefy Lanza (nextime / spora )
Browse files
Options

Fix Wan 2.2 I2V base model detection 

- Fixed mapping table to use correct I2V base model (Wan-AI/Wan2.2-I2V-14B-Diffusers)
- Fixed Diffuser -> Diffusers typo in model IDs
- Updated all Wan 2.2 I2V references to use correct model ID
parent d5ac0826
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
videogen.py
View file @ 4a5213f8
......@@ -1431,7 +1431,7 @@ def add_model_from_hf(model_id_or_url, name=None, hf_token=None, debug=False):
if "wan" in model_id.lower():
if "wan2.2" in model_id.lower():
# Wan 2.2 models - use the new MoE base
base_model = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if is_i2v else "Wan-AI/Wan2.2-T2V-14B-Diffuser"
base_model = "Wan-AI/Wan2.2-I2V-14B-Diffusers" if is_i2v else "Wan-AI/Wan2.2-T2V-14B-Diffusers"
else:
# Wan 2.1 and earlier
base_model = "Wan-AI/Wan2.1-I2V-14B-Diffusers" if is_i2v else "Wan-AI/Wan2.1-T2V-14B-Diffusers"
......@@ -3868,7 +3868,7 @@ def select_best_model(gen_type, models, vram_gb=24, prefer_quality=True, return_
if "wan" in lora_id:
if "wan2.2" in lora_id:
# Wan 2.2 models - use the new MoE base
base_model_id = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if info.get("supports_i2v") else "Wan-AI/Wan2.2-T2V-14B-Diffuser"
base_model_id = "Wan-AI/Wan2.2-I2V-14B-Diffusers" if info.get("supports_i2v") else "Wan-AI/Wan2.2-T2V-14B-Diffusers"
else:
# Wan 2.1 and earlier
base_model_id = "Wan-AI/Wan2.1-I2V-14B-Diffusers" if info.get("supports_i2v") else "Wan-AI/Wan2.1-T2V-14B-Diffusers"
......@@ -9214,9 +9214,9 @@ def main(args):
base_model_fallbacks = {
"ltx": "Lightricks/LTX-Video",
"ltxvideo": "Lightricks/LTX-Video",
# Wan 2.2 models - use T2V base for all (I2V models use same base)
"wan2.2.i2v.a14b": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
"wan2.2.i2v": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
# Wan 2.2 models - I2V uses I2V base, T2V uses T2V base
"wan2.2.i2v.a14b": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
"wan2.2.i2v": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
"wan2.2": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
# Wan 2.1 models
"wan2.1.i2v.a14b": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment