From 177f8f526f323b06f7322bc3cfb5e31efb7e3665 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ma=C3=ABlys=20Jusseaux?= <https://gitlab.com/mjusseaux>
Date: Fri, 26 Jul 2024 10:49:27 +0200
Subject: [PATCH] corrected ImageInference node name

---
 diffusers_in_comfy.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/diffusers_in_comfy.py b/diffusers_in_comfy.py
index a54bb50..06c58cb 100644
--- a/diffusers_in_comfy.py
+++ b/diffusers_in_comfy.py
@@ -283,7 +283,7 @@ class BLoRALoader:
 
 NODE_CLASS_MAPPINGS = {
     "CreatePipeline": GenerateStableDiffusionPipeline,
-    "GenerateImage": ImageInference,
+    "ImageInference": ImageInference,
     "LoRALoader" : LoRALoader,
     "BLoRALoader" : BLoRALoader,
     "MakeCanny": MakeCanny,
@@ -292,7 +292,7 @@ NODE_CLASS_MAPPINGS = {
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "CreatePipeline" : "GenerateStableDiffusionPipeline",
-    "GenerateImage" : "ImageInference",
+    "ImageInference" : "ImageInference",
     "LoRALoader" : "LoRALoader",
     "BLoRALoader" : "BLoRALoader",
     "MakeCanny": "MakeCanny",