From c906c5d14a7df7f6c8819f1698cc1dc2b7bf92df Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Sat, 29 Jul 2023 09:14:00 +0530 Subject: [PATCH] Don't rely on old keys to exist in the request --- ui/easydiffusion/types.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ui/easydiffusion/types.py b/ui/easydiffusion/types.py index b5f6b21a..a37da9ef 100644 --- a/ui/easydiffusion/types.py +++ b/ui/easydiffusion/types.py @@ -217,11 +217,11 @@ def convert_legacy_render_req_to_new(old_req: dict): # move the model params if model_paths["stable-diffusion"]: - model_params["stable-diffusion"] = {"clip_skip": bool(old_req["clip_skip"])} + model_params["stable-diffusion"] = {"clip_skip": bool(old_req.get("clip_skip", False))} # move the filter params if model_paths["realesrgan"]: - filter_params["realesrgan"] = {"scale": int(old_req["upscale_amount"])} + filter_params["realesrgan"] = {"scale": int(old_req.get("upscale_amount", 4))} if model_paths["latent_upscaler"]: filter_params["latent_upscaler"] = { "prompt": old_req["prompt"], @@ -232,8 +232,8 @@ def convert_legacy_render_req_to_new(old_req: dict): } if model_paths["codeformer"]: filter_params["codeformer"] = { - "upscale_faces": bool(old_req["codeformer_upscale_faces"]), - "codeformer_fidelity": float(old_req["codeformer_fidelity"]), + "upscale_faces": bool(old_req.get("codeformer_upscale_faces", True)), + "codeformer_fidelity": float(old_req.get("codeformer_fidelity", 0.5)), } # set the filters