From 3ac4acbf78a8da42bbcb001938bd148d2f736397 Mon Sep 17 00:00:00 2001
From: Fayche <1751162157@qq.com>
Date: Tue, 17 Dec 2024 15:12:09 +0800
Subject: [PATCH] fix: Repair ComfyUI/issues/6074

---
 node.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/node.py b/node.py
index 596ec55..7703239 100644
--- a/node.py
+++ b/node.py
@@ -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