fixed i2v 720p no longer recognized properly

This commit is contained in:
deepbeepmeep 2025-06-13 14:26:45 +02:00 committed by GitHub
parent 5a63326bb9
commit 85e488f76c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

2
wgp.py
View File

@ -1624,7 +1624,7 @@ def get_model_family(model_type):
def test_class_i2v(model_type):
model_type = get_base_model_type(model_type)
return model_type in ["i2v", "fun_inp_1.3B", "fun_inp", "flf2v_720p", "fantasy", "hunyuan_i2v" ]
return model_type in ["i2v", "i2v_720p", "fun_inp_1.3B", "fun_inp", "flf2v_720p", "fantasy", "hunyuan_i2v" ]
def get_model_name(model_type, description_container = [""]):
finetune_def = get_model_finetune_def(model_type)