Skip to content

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

fix: Use Wan2.2-T2V as base for all Wan 2.2 models 

The user confirmed that Wan2.2-I2V models should use Wan2.2-T2V
as the base model, not the I2V variant.
parent 77a089f8
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
videogen.py
View file @ 1162d3c0
......@@ -9202,13 +9202,13 @@ def main(args):
base_model_fallbacks = {
"ltx": "Lightricks/LTX-Video",
"ltxvideo": "Lightricks/LTX-Video",
# Wan models - more specific keys MUST come first!
# (longer keys checked first so wan2.2.i2v matches before wan2.2)
"wan2.2.i2v.a14b": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
"wan2.2.i2v": "Wan-AI/Wan2.2-I2V-14B-Diffusers",
# 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",
"wan2.2": "Wan-AI/Wan2.2-T2V-14B-Diffusers",
"wan2.1.i2v.a14b": "Wan-AI/Wan2.1-I2V-14B-Diffusers",
"wan2.1.i2v": "Wan-AI/Wan2.1-I2V-14B-Diffusers",
# Wan 2.1 models
"wan2.1.i2v.a14b": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
"wan2.1.i2v": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
"wan2.1": "Wan-AI/Wan2.1-T2V-14B-Diffusers",
# Generic Wan fallback (least specific - checked last)
"wan": "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