From 85e488f76cd18e12deb3e6221b100fcc8642b3c7 Mon Sep 17 00:00:00 2001
From: deepbeepmeep <84379123+deepbeepmeep@users.noreply.github.com>
Date: Fri, 13 Jun 2025 14:26:45 +0200
Subject: [PATCH] fixed i2v 720p no longer recognized properly

---
 wgp.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/wgp.py b/wgp.py
index d9e6b43..15d35dc 100644
--- a/wgp.py
+++ b/wgp.py
@@ -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)
@@ -6252,4 +6252,4 @@ if __name__ == "__main__":
         else:
             url = "http://" + server_name 
         webbrowser.open(url + ":" + str(server_port), new = 0, autoraise = True)
-    demo.launch(server_name=server_name, server_port=server_port, share=args.share, allowed_paths=[save_path])
\ No newline at end of file
+    demo.launch(server_name=server_name, server_port=server_port, share=args.share, allowed_paths=[save_path])