From 67ce39717c56fd2902e9014d7010b9a9f720ea3b Mon Sep 17 00:00:00 2001
From: Gius <112352961+GiusTex@users.noreply.github.com>
Date: Thu, 15 May 2025 16:06:52 +0200
Subject: [PATCH] Update __init__.py

---
 __init__.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index c789f9e..07f918c 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,15 +1,17 @@
-from .nodes import (PadImageForDiffusersOutpaint, LoadDiffusersOutpaintModels, EncodeDiffusersOutpaintPrompt, DiffusersImageOutpaint)
+from .nodes import (PadImageForDiffusersOutpaint, LoadDiffuserModel, LoadDiffuserControlnet, EncodeDiffusersOutpaintPrompt, DiffusersImageOutpaint)
 
 NODE_CLASS_MAPPINGS = {
     "PadImageForDiffusersOutpaint": PadImageForDiffusersOutpaint,
-    "LoadDiffusersOutpaintModels": LoadDiffusersOutpaintModels,
+    "LoadDiffuserModel": LoadDiffuserModel,
+    "LoadDiffuserControlnet": LoadDiffuserControlnet,
     "EncodeDiffusersOutpaintPrompt": EncodeDiffusersOutpaintPrompt,
     "DiffusersImageOutpaint": DiffusersImageOutpaint
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "PadImageForDiffusersOutpaint": "Pad Image For Diffusers Outpaint",
-    "LoadDiffusersOutpaintModels": "Load Diffusers Outpaint Models",
+    "LoadDiffuserModel": "Load Diffuser Model",
+    "LoadDiffuserControlnet": "Load Diffuser Controlnet",
     "EncodeDiffusersOutpaintPrompt": "Encode Diffusers Outpaint Prompt",
     "DiffusersImageOutpaint": "Diffusers Image Outpaint"
 }