StableDiffusionControlNetPipeline

handler.py

@@ -1,13 +1,13 @@
 from typing import Dict, List, Any
-from diffusers import StableDiffusionPipeline
+from diffusers import (
+    ControlNetModel,
+    StableDiffusionControlNetPipeline,
+)
 
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.pipeline = StableDiffusionPipeline.from_single_file(
-            "https://huggingface.co/artificialguybr/LogoRedmond-LogoLoraForSDXL/blob/main/LogoRedmond_LogoRedAF.safetensors",
-            use_safetensors=True
-        )
+        self.pipeline = StableDiffusionControlNetPipeline.from_single_file("https://huggingface.co/artificialguybr/LogoRedmond-LogoLoraForSDXL/blob/main/LogoRedmond_LogoRedAF.safetensors")
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """