diff --git a/simpletuner/helpers/models/wan/model.py b/simpletuner/helpers/models/wan/model.py index edfa1bcdf..7f5ce574b 100644 --- a/simpletuner/helpers/models/wan/model.py +++ b/simpletuner/helpers/models/wan/model.py @@ -225,8 +225,8 @@ class Wan(VideoModelFoundation): "t2v-480p-14b-2.1": "Wan-AI/Wan2.1-T2V-14B-Diffusers", "i2v-14b-2.1": "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers", "i2v-14b-2.1-720p": "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers", - "i2v-14b-2.2-high": "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers", - "i2v-14b-2.2-low": "Wan-AI/Wan2.1-I2V-14B-720P-Diffusers", + "i2v-14b-2.2-high": "Wan-AI/Wan2.2-I2V-A14B-Diffusers", + "i2v-14b-2.2-low": "Wan-AI/Wan2.2-I2V-A14B-Diffusers", "flf2v-14b-2.1": "Wan-AI/Wan2.1-FLF2V-14B-720P-diffusers", "flf2v-14b-2.2-high": "Wan-AI/Wan2.1-FLF2V-14B-720P-diffusers", "flf2v-14b-2.2-low": "Wan-AI/Wan2.1-FLF2V-14B-720P-diffusers",