Skip to content

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

Fix Wan 2.2 base model name: Diffusers (with s)

parent 2d9d09f6
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
videogen.py
View file @ d5ac0826
......@@ -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-Diffuser" if is_i2v else "Wan-AI/Wan2.2-T2V-14B-Diffuser"
base_model = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if is_i2v else "Wan-AI/Wan2.2-T2V-14B-Diffuser"
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-Diffuser" if info.get("supports_i2v") else "Wan-AI/Wan2.2-T2V-14B-Diffuser"
base_model_id = "Wan-AI/Wan2.2-I2V-A14B-Diffusers" if info.get("supports_i2v") else "Wan-AI/Wan2.2-T2V-14B-Diffuser"
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"
......
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