diff --git a/__init__.py b/__init__.py index 80c74dd..7ac3e50 100644 --- a/__init__.py +++ b/__init__.py @@ -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" -} \ No newline at end of file + "KY_SaveImageToPath": "Save Image To target Path", + "KY_OpenAICaptionImage": "Describe Image from openai-protocol" +}