Update handler.py

handler.py

@@ -8,7 +8,7 @@ class EndpointHandler:
     def __init__(self, path: str = ""):
         print(f"Initializing SDXL model from: {path}")
 
-        # Base SDXL Model
+        # Load the base SDXL model
         self.pipe = StableDiffusionXLPipeline.from_pretrained(
             "stabilityai/stable-diffusion-xl-base-1.0",
             torch_dtype=torch.float16,
@@ -16,7 +16,12 @@ class EndpointHandler:
         )
 
         print("Loading LoRA weights from: Texttra/Bh0r")
-        self.pipe.load_lora_weights("Texttra/Bh0r", weight_name="Bh0r-10.safetensors")
+        self.pipe.load_lora_weights(
+            "Texttra/Bh0r",
+            weight_name="Bh0r-10.safetensors",
+            adapter_name="bh0r_lora"
+        )
+        self.pipe.set_adapters(["bh0r_lora"], adapter_weights=[0.9])
         self.pipe.fuse_lora()
 
         self.pipe.to("cuda" if torch.cuda.is_available() else "cpu")