Mirza Učanbarlić commited on
Commit ·
c058aa4
1
Parent(s): 71ffd5c
add cuda
Browse files- handler.py +1 -1
handler.py
CHANGED
|
@@ -10,7 +10,7 @@ from diffusers.utils import load_image
|
|
| 10 |
class EndpointHandler():
|
| 11 |
def __init__(self, path=""):
|
| 12 |
repo_id = "runwayml/stable-diffusion-v1-5"
|
| 13 |
-
self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id)
|
| 14 |
weight_name = "pixel-portrait-v1.safetensors"
|
| 15 |
self.pipeline.load_lora_weights("simulationcartridge/ppl", weight_name=weight_name)
|
| 16 |
|
|
|
|
| 10 |
class EndpointHandler():
|
| 11 |
def __init__(self, path=""):
|
| 12 |
repo_id = "runwayml/stable-diffusion-v1-5"
|
| 13 |
+
self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id).to("cuda")
|
| 14 |
weight_name = "pixel-portrait-v1.safetensors"
|
| 15 |
self.pipeline.load_lora_weights("simulationcartridge/ppl", weight_name=weight_name)
|
| 16 |
|