handler.py

handler.py

@@ -206,7 +206,7 @@ class EndpointHandler():
                  #    print("get prompt from request: ", prompt)
                  #    args["prompt"] = prompt
                 p = StableDiffusionProcessingTxt2Img(sd_model=self.shared.sd_model, **txt2img_args)
-        if data["type"] == "txt2img":
+        if data["type"] == "img2img":
             if data["inputs"]:
                 for field in img2img_args:
                     if field in data["inputs"].keys():