diff --git a/__init__.py b/__init__.py index a5cce89..c789f9e 100644 --- a/__init__.py +++ b/__init__.py @@ -1,13 +1,15 @@ -from .nodes import (PadImageForDiffusersOutpaint, LoadDiffusersOutpaintModels, DiffusersImageOutpaint) +from .nodes import (PadImageForDiffusersOutpaint, LoadDiffusersOutpaintModels, EncodeDiffusersOutpaintPrompt, DiffusersImageOutpaint) NODE_CLASS_MAPPINGS = { "PadImageForDiffusersOutpaint": PadImageForDiffusersOutpaint, "LoadDiffusersOutpaintModels": LoadDiffusersOutpaintModels, + "EncodeDiffusersOutpaintPrompt": EncodeDiffusersOutpaintPrompt, "DiffusersImageOutpaint": DiffusersImageOutpaint } NODE_DISPLAY_NAME_MAPPINGS = { "PadImageForDiffusersOutpaint": "Pad Image For Diffusers Outpaint", "LoadDiffusersOutpaintModels": "Load Diffusers Outpaint Models", + "EncodeDiffusersOutpaintPrompt": "Encode Diffusers Outpaint Prompt", "DiffusersImageOutpaint": "Diffusers Image Outpaint" -} \ No newline at end of file +}