fix(handler): remove fp16
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -17,7 +17,7 @@ class EndpointHandler():
|
|
17 |
# load the optimized model
|
18 |
import torch
|
19 |
|
20 |
-
self.pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo"
|
21 |
self.pipe.load_lora_weights("ppierzc/sd-icongenie-model-lora-sdxl", weight_name="pytorch_lora_weights.safetensors")
|
22 |
|
23 |
self.pipe = self.pipe.to(device)
|
|
|
17 |
# load the optimized model
|
18 |
import torch
|
19 |
|
20 |
+
self.pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
|
21 |
self.pipe.load_lora_weights("ppierzc/sd-icongenie-model-lora-sdxl", weight_name="pytorch_lora_weights.safetensors")
|
22 |
|
23 |
self.pipe = self.pipe.to(device)
|