feat: enhance GPU capability checks and refactor configuration management in bridge.py; add support for model class list in model_utils.py; update inference and memory optimization configurations in nodes.py

2025-09-04 22:36:37 +08:00
parent 40bbc722e7
commit 4e75a5dc0c
4 changed files with 369 additions and 356 deletions
@@ -29,6 +29,22 @@ def scan_models() -> List[str]:
    return ["None"] + models if models else ["None"]


+def support_model_cls_list(model_cls: str) -> List[str]:
+    return [
+        "wan2.1",
+        "wan2.1_distill",
+        "wan2.1_vace",
+        "cogvideox",
+        "seko_talk",
+        "wan2.2_moe",
+        "wan2.2",
+        "wan2.2_moe_audio",
+        "wan2.2_audio",
+        "wan2.2_moe_distill",
+        "qwen_image",
+    ]
+
+
 def scan_loras() -> List[str]:
    loras = []
    base_path = get_model_base_path()