Oysiyl
/

sd-lora-android-google-toy

Inference Endpoints

Model card Files Files and versions Community

Oysiyl commited on Feb 1

Commit

b1f160a

•

1 Parent(s): 8736bbd

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -22,7 +22,7 @@ class EndpointHandler():
         self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
         self.pipe = AutoPipelineForText2Image.from_pretrained(self.stable_diffusion_id,
                                                               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

         self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
         self.pipe = AutoPipelineForText2Image.from_pretrained(self.stable_diffusion_id,
                                                               torch_dtype=dtype)
+        self.pipe.load_lora_weights(".", weights="pytorch_lora_weights.safetensors")
         self.pipe.enable_xformers_memory_efficient_attention()
         self.pipe = self.pipe.to(device)
         self.seed = 42