Update __init__.py

2024-12-03 14:59:35 +08:00
parent f1acec0c32
commit 6024765678
1 changed files with 10 additions and 3 deletions
@@ -1,13 +1,20 @@
 from .advanced_lying_sigma_sampler import AdvancedLyingSigmaSamplerNode
 from .KY_SaveImageToPath import KY_SaveImageToPath
+from .openai_VLM import OpenAICaptionImage
+
+NODE_CLASS_MAPPINGS = {
+    "OpenAI.CaptionImage": OpenAICaptionImage
+}

 # 将节点映射到NODE_CLASS_MAPPINGS和NODE_DISPLAY_NAME_MAPPINGS
 NODE_CLASS_MAPPINGS = {
    "AdvancedLyingSigmaSampler": AdvancedLyingSigmaSamplerNode,
-    "KY_SaveImageToPath": KY_SaveImageToPath
+    "KY_SaveImageToPath": KY_SaveImageToPath,
+    "OpenAICaptionImage": KY_OpenAICaptionImage
 }

 NODE_DISPLAY_NAME_MAPPINGS = {
    "AdvancedLyingSigmaSampler": "Advanced Lying Sigma Sampler",
-    "KY_SaveImageToPath": "Save Image To target Path"
-}
+    "KY_SaveImageToPath": "Save Image To target Path",
+    "KY_OpenAICaptionImage": "Describe Image from openai-protocol"
+}