Skip to content
Commit ec05b598 authored by Stefy Lanza (nextime / spora )'s avatar Stefy Lanza (nextime / spora )
Browse files

fix: Detect Wan 2.2 I2V base model correctly for image-to-video models

Before this fix, using Wan 2.2 I2V models like 'wan2_2_i2v_a14b' would
incorrectly try to load 'Wan-AI/Wan2.1-T2V-14B-Diffusers' as the base
model because the fallback logic only matched 'wan' generically.

Now the base_model_fallbacks dictionary includes specific entries for:
- Wan 2.2 I2V models: wan2_2_i2v, wan2.2_i2v
- Wan 2.2 T2V models: wan2_2, wan2.2
- Wan 2.1 I2V models: wan2_1_i2v, wan2.1_i2v
- Wan 2.1 T2V models: wan2_1, wan2.1
- Generic Wan fallback: wan

The more specific keys are checked first, so model IDs like
'wan2_2_i2v_a14b' will correctly match 'wan2_2_i2v' and use
'Wan-AI/Wan2.2-I2V-14B-Diffusers' as the base model.
parent fbefb047
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment