extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-01 11:37:04 +02:00
Code Issues Packages Projects Releases Wiki Activity

Don't rely on old keys to exist in the request

Browse Source
This commit is contained in:
cmdr2 2023-07-29 09:14:00 +05:30
parent 6e52680fa8
commit c906c5d14a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ui/easydiffusion/types.py
View File

@ -217,11 +217,11 @@ def convert_legacy_render_req_to_new(old_req: dict):
# move the model params # move the model params
if model_paths["stable-diffusion"]: 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 # move the filter params
if model_paths["realesrgan"]: 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"]: if model_paths["latent_upscaler"]:
filter_params["latent_upscaler"] = { filter_params["latent_upscaler"] = {
"prompt": old_req["prompt"], "prompt": old_req["prompt"],
@ -232,8 +232,8 @@ def convert_legacy_render_req_to_new(old_req: dict):
} }
if model_paths["codeformer"]: if model_paths["codeformer"]:
filter_params["codeformer"] = { filter_params["codeformer"] = {
"upscale_faces": bool(old_req["codeformer_upscale_faces"]), "upscale_faces": bool(old_req.get("codeformer_upscale_faces", True)),
"codeformer_fidelity": float(old_req["codeformer_fidelity"]), "codeformer_fidelity": float(old_req.get("codeformer_fidelity", 0.5)),
} }
# set the filters # set the filters