simulationcartridge
/

ppl

stable-diffusion

template:sd-lora

Model card Files Files and versions

Mirza Učanbarlić commited on Nov 11, 2023

Commit

dcdeaa3

·

1 Parent(s): 510acad

fix typo

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,7 +26,7 @@ class EndpointHandler():
         prompt = data.pop("prompt", None)
         # run normal prediction
-        output = self.pipeline(prompt=prompt, image=init_image)
         image = output.images[0]

         prompt = data.pop("prompt", None)
         # run normal prediction
+        output = self.pipeline(prompt=prompt, image=input_image)
         image = output.images[0]