Update handler.py

handler.py

@@ -21,11 +21,10 @@ class EndpointHandler():
         # Load StableDiffusionPipeline 
         self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
         self.pipe = AutoPipelineForText2Image.from_pretrained(self.stable_diffusion_id, 
-                                                              torch_dtype=dtype,
-                                                              safety_checker=None)
-        pipe.load_lora_weights("pytorch_lora_weights.safetensors")
-        pipe.enable_xformers_memory_efficient_attention()
-        pipe = pipe.to(device)
+                                                              torch_dtype=dtype)
+        self.pipe.load_lora_weights("pytorch_lora_weights.safetensors")
+        self.pipe.enable_xformers_memory_efficient_attention()
+        self.pipe = self.pipe.to(device)
         self.seed = 42
         # Define Generator with seed
         self.generator = torch.Generator(device="cpu").manual_seed(self.seed)