ppierzc
/

sd-icongenie-model-lora-sdxl

stable-diffusion-xl

stable-diffusion-xl-diffusers

Model card Files Files and versions Community

ppierzc commited on Dec 21, 2023

Commit

fe29560

·

1 Parent(s): 74ff852

fix(handler)

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -18,8 +18,8 @@ class EndpointHandler():
         import torch
         self.pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
-        print(path)
-        self.pipe.load_lora_weights(path, weight_name="pytorch_lora_weights.safetensors")
         self.pipe = self.pipe.to(device)

         import torch
         self.pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16")
+        print("path", path)
+        self.pipe.load_lora_weights("ppierzc/sd-icongenie-model-lora-sdxl", weight_name="pytorch_lora_weights.safetensors")
         self.pipe = self.pipe.to(device)