simulationcartridge
/

ppl

stable-diffusion

template:sd-lora

Model card Files Files and versions

Mirza Učanbarlić commited on Nov 11, 2023

Commit

c2c212c

·

1 Parent(s): 09fe324

add missing self

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -8,7 +8,7 @@ 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]]:
         """

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