fix: Repair ComfyUI/issues/6074

2024-12-17 15:12:09 +08:00
parent 00348f0a3d
commit 3ac4acbf78
1 changed files with 6 additions and 1 deletions
@@ -3,6 +3,9 @@ from dashscope import Generation


 class DashscopeLLMLoader:
+    def __init__(self):
+        pass
+
    @classmethod
    def INPUT_TYPES(cls):
        return {
@@ -51,10 +54,12 @@ class DashscopeLLMLoader:
            }
        }

-    CATEGORY = "dashscope"
    FUNCTION = "select_model"
+    OUTPUT_NODE = True
    RETURN_TYPES = ("String",)

+    CATEGORY = "dashscope"
+
    def select_model(self, model_type, qwen_max, qwen_plus, qwen_turbo, qwen_long):
        if model_type == "qwen_max":
            model_version = qwen_max