simulationcartridge
/

ppl

stable-diffusion

template:sd-lora

Model card Files Files and versions

Mirza Učanbarlić commited on Nov 11, 2023

Commit

c058aa4

·

1 Parent(s): 71ffd5c

add cuda

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -10,7 +10,7 @@ from diffusers.utils import load_image
 class EndpointHandler():
     def __init__(self, path=""):
         repo_id = "runwayml/stable-diffusion-v1-5"
-        self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id)
         weight_name = "pixel-portrait-v1.safetensors"
         self.pipeline.load_lora_weights("simulationcartridge/ppl", weight_name=weight_name)

 class EndpointHandler():
     def __init__(self, path=""):
         repo_id = "runwayml/stable-diffusion-v1-5"
+        self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id).to("cuda")
         weight_name = "pixel-portrait-v1.safetensors"
         self.pipeline.load_lora_weights("simulationcartridge/ppl", weight_name=weight_name)