From 56677782ccff8a5e20862d6fe0c9fd1f69b4eaa2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ma=C3=ABlys=20Jusseaux?= <https://gitlab.com/mjusseaux>
Date: Thu, 25 Jul 2024 18:08:34 +0200
Subject: [PATCH] renamed pipeline node to avoid confusion

---
 diffusers_in_comfy.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/diffusers_in_comfy.py b/diffusers_in_comfy.py
index 0c73978..d77dd44 100644
--- a/diffusers_in_comfy.py
+++ b/diffusers_in_comfy.py
@@ -15,7 +15,7 @@ from PIL import Image
 
 
 
-class StableDiffusionPipeline:
+class GenerateStableDiffusionPipeline:
     def __init__(self) -> None:
         pass
 
@@ -281,7 +281,7 @@ class BLoRALoader:
     
 
 NODE_CLASS_MAPPINGS = {
-    "CreatePipeline": StableDiffusionPipeline,
+    "CreatePipeline": GenerateStableDiffusionPipeline,
     "GenerateImage": ImageInference,
     "LoRALoader" : LoRALoader,
     "BLoRALoader" : BLoRALoader,
@@ -290,7 +290,7 @@ NODE_CLASS_MAPPINGS = {
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "CreatePipeline" : "StableDiffusionPipeline",
+    "CreatePipeline" : "GenerateStableDiffusionPipeline",
     "GenerateImage" : "ImageInference",
     "LoRALoader" : "LoRALoader",
     "BLoRALoader" : "BLoRALoader",