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"
 }