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
...@@ -1431,7 +1431,7 @@ def add_model_from_hf(model_id_or_url, name=None, hf_token=None, debug=False): ...@@ -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 "wan" in model_id.lower():
if "wan2.2" in model_id.lower(): if "wan2.2" in model_id.lower():
# Wan 2.2 models - use the new MoE base # 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: else:
# Wan 2.1 and earlier # 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" 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_ ...@@ -3868,7 +3868,7 @@ def select_best_model(gen_type, models, vram_gb=24, prefer_quality=True, return_
if "wan" in lora_id: if "wan" in lora_id:
if "wan2.2" in lora_id: if "wan2.2" in lora_id:
# Wan 2.2 models - use the new MoE base # 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: else:
# Wan 2.1 and earlier # 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" 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): ...@@ -9214,9 +9214,9 @@ def main(args):
base_model_fallbacks = { base_model_fallbacks = {
"ltx": "Lightricks/LTX-Video", "ltx": "Lightricks/LTX-Video",
"ltxvideo": "Lightricks/LTX-Video", "ltxvideo": "Lightricks/LTX-Video",
# Wan 2.2 models - use T2V base for all (I2V models use same base) # Wan 2.2 models - I2V uses I2V base, T2V uses T2V base
"wan2.2.i2v.a14b": "Wan-AI/Wan2.2-T2V-14B-Diffusers", "wan2.2.i2v.a14b": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
"wan2.2.i2v": "Wan-AI/Wan2.2-T2V-14B-Diffusers", "wan2.2.i2v": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
"wan2.2": "Wan-AI/Wan2.2-T2V-14B-Diffusers", "wan2.2": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
# Wan 2.1 models # Wan 2.1 models
"wan2.1.i2v.a14b": "Wan-AI/Wan2.1-T2V-14B-Diffusers", "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