safetensors load_lora_weights

handler.py

@@ -1,12 +1,15 @@
 from typing import Dict, List, Any
-from diffusers import ControlNetModel, StableDiffusionControlNetPipeline, StableDiffusionPipeline
+from diffusers import ControlNetModel, StableDiffusionControlNetPipeline, StableDiffusionPipeline, DiffusionPipeline
 
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.pipeline = StableDiffusionPipeline.from_single_file(
-            "https://huggingface.co/artificialguybr/LogoRedmond-LogoLoraForSDXL/blob/main/LogoRedmond_LogoRedAF.safetensors"
-        )
+        self.pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
+        self.pipeline.load_lora_weights(".", weight_name="./safetensors.safetensors")
+
+        # self.pipeline = StableDiffusionPipeline.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]]:
         """