simulationcartridge
/

ppl

stable-diffusion

template:sd-lora

Model card Files Files and versions

Mirza Učanbarlić commited on Nov 11, 2023

Commit

09fe324

·

1 Parent(s): 64b6937

remove cuda

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -7,7 +7,7 @@ from io import BytesIO
 class EndpointHandler():
     def __init__(self, path=""):
         repo_id = "runwayml/stable-diffusion-v1-5"
-        self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id).to("cuda")
         pipeline.load_lora_weights("simulationcartridge/ppl")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:

 class EndpointHandler():
     def __init__(self, path=""):
         repo_id = "runwayml/stable-diffusion-v1-5"
+        self.pipeline = StableDiffusionImg2ImgPipeline.from_pretrained(repo_id)
         pipeline.load_lora_weights("simulationcartridge/ppl")
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]: