Skip to content

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

Fix Wan 2.2 base model IDs to use A14B suffix

parent 6a982a8b
Show whitespace changes
Inline Side-by-side
Showing with 13 additions and 13 deletions
videogen.py
View file @ 1f33b9e5
......@@ -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-14B-Diffusers" if is_i2v else "Wan-AI/Wan2.2-T2V-14B-Diffusers"
base_model = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if is_i2v else "Wan-AI/Wan2.2-T2V-A14B-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"
......@@ -2241,7 +2241,7 @@ def update_all_models(hf_token=None):
if "wan" in model_id.lower():
# Wan 2.2 models - use the new MoE base
if "wan2.2" in model_id.lower() or "wan2_2" in model_id.lower():
base_model = "Wan-AI/Wan2.2-I2V-14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.2-T2V-14B-Diffusers"
base_model = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
else:
# Wan 2.1 and earlier
base_model = "Wan-AI/Wan2.1-I2V-14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.1-T2V-14B-Diffusers"
......@@ -2347,7 +2347,7 @@ def update_all_models(hf_token=None):
if "wan" in model_id.lower():
# Wan 2.2 models - use the new MoE base
if "wan2.2" in model_id.lower() or "wan2_2" in model_id.lower():
base_model = "Wan-AI/Wan2.2-I2V-14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.2-T2V-14B-Diffusers"
base_model = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
else:
# Wan 2.1 and earlier
base_model = "Wan-AI/Wan2.1-I2V-14B-Diffusers" if m["is_i2v"] else "Wan-AI/Wan2.1-T2V-14B-Diffusers"
......@@ -3879,7 +3879,7 @@ def select_best_model(gen_type, models, vram_gb=24, prefer_quality=True, return_
if "wan2.2" in lora_id:
# Wan 2.2 models - use the new MoE base
# IMPORTANT: For I2V models, always use I2V base model, not T2V
base_model_id = "Wan-AI/Wan2.2-I2V-14B-Diffusers" if "i2v" in lora_id else "Wan-AI/Wan2.2-T2V-14B-Diffusers"
base_model_id = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if "i2v" in lora_id else "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
else:
# Wan 2.1 and earlier
base_model_id = "Wan-AI/Wan2.1-I2V-14B-Diffusers" if "i2v" in lora_id else "Wan-AI/Wan2.1-T2V-14B-Diffusers"
......@@ -8084,9 +8084,9 @@ def main(args):
# Wan 2.2 models - use lora_id_lower to determine I2V vs T2V
# This is more reliable than m_info.get("supports_i2v")
if "i2v" in lora_id_lower:
base_model_id = "Wan-AI/Wan2.2-I2V-14B-Diffusers"
base_model_id = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
else:
base_model_id = "Wan-AI/Wan2.2-T2V-14B-Diffusers"
base_model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
elif "wan2.1" in lora_id_lower or "wan2_1" in lora_id_lower:
# Wan 2.1 models - use lora_id_lower to determine I2V vs T2V
if "i2v" in lora_id_lower:
......@@ -8097,9 +8097,9 @@ def main(args):
# Generic Wan - check the lora_id for i2v instead of m_info
# This is more reliable as m_info.supports_i2v may not be set correctly
if "i2v" in lora_id_lower:
base_model_id = "Wan-AI/Wan2.2-I2V-14B-Diffusers"
base_model_id = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
else:
base_model_id = "Wan-AI/Wan2.2-T2V-14B-Diffusers"
base_model_id = "Wan-AI/Wan2.2-T2V-A14B-Diffusers"
elif "svd" in lora_id_lower or "stable-video" in lora_id_lower:
base_model_id = "stabilityai/stable-video-diffusion-img2vid-xt-1-1"
elif "sdxl" in lora_id_lower:
......@@ -9229,11 +9229,11 @@ def main(args):
"ltx": "Lightricks/LTX-Video",
"ltxvideo": "Lightricks/LTX-Video",
# Wan 2.2 I2V models - more specific keys FIRST (before generic "wan2.2")
"wan2.2.i2v.a14b": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
"wan2.2.i2v": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
"wan2.2.t2v": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
"wan2.2.i2v.a14b": "Wan-AI/Wan2.2-I2V-A14B-Diffusers",
"wan2.2.i2v": "Wan-AI/Wan2.2-I2V-A14B-Diffusers",
"wan2.2.t2v": "Wan-AI/Wan2.2-T2V-A14B-Diffusers",
# Wan 2.2 generic - MUST come after specific I2V/T2V keys
"wan2.2": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
"wan2.2": "Wan-AI/Wan2.2-T2V-A14B-Diffusers",
# Wan 2.1 I2V models - more specific keys FIRST
"wan2.1.i2v.a14b": "Wan-AI/Wan2.1-I2V-14B-Diffusers",
"wan2.1.i2v": "Wan-AI/Wan2.1-I2V-14B-Diffusers",
......@@ -9241,7 +9241,7 @@ def main(args):
# Wan 2.1 generic - MUST come after specific I2V/T2V keys
"wan2.1": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
# Generic Wan fallback (least specific - checked last)
"wan": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
"wan": "Wan-AI/Wan2.2-T2V-A14B-Diffusers",
"svd": "stabilityai/stable-video-diffusion-img2vid-xt-1-1",
"cogvideo": "THUDM/CogVideoX-5b",
"mochi": "genmo/mochi-1-preview",
......
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