From 3b923e0d370a14150ad7e68de9b3605db406ac08 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Sat, 8 Oct 2022 12:48:37 +0530 Subject: [PATCH] Didn't fix the ckpt bug properly --- ui/sd_internal/runtime.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/ui/sd_internal/runtime.py b/ui/sd_internal/runtime.py index 8e0657a6..c17d7cef 100644 --- a/ui/sd_internal/runtime.py +++ b/ui/sd_internal/runtime.py @@ -224,8 +224,9 @@ def do_mk_img(req: Request): # to the ckpt file (without the extension). needs_model_reload = False - if ckpt_file != req.use_stable_diffusion_model: - ckpt_file = req.use_stable_diffusion_model + ckpt_to_use = ckpt_file + if ckpt_to_use != req.use_stable_diffusion_model: + ckpt_to_use = req.use_stable_diffusion_model needs_model_reload = True model.turbo = req.turbo @@ -235,7 +236,7 @@ def do_mk_img(req: Request): if model_is_half: del model, modelCS, modelFS - load_model_ckpt(ckpt_file, device) + load_model_ckpt(ckpt_to_use, device) needs_model_reload = False load_model_gfpgan(gfpgan_file) @@ -249,7 +250,7 @@ def do_mk_img(req: Request): (precision == 'full' and not req.use_full_precision and not force_full_precision): del model, modelCS, modelFS - load_model_ckpt(ckpt_file, device, req.turbo, unet_bs, ('full' if req.use_full_precision else 'autocast')) + load_model_ckpt(ckpt_to_use, device, req.turbo, unet_bs, ('full' if req.use_full_precision else 'autocast')) needs_model_reload = False if prev_device != device: @@ -257,7 +258,7 @@ def do_mk_img(req: Request): load_model_real_esrgan(real_esrgan_file) if needs_model_reload: - load_model_ckpt(ckpt_file, device, req.turbo, unet_bs, precision) + load_model_ckpt(ckpt_to_use, device, req.turbo, unet_bs, precision) if req.use_face_correction != gfpgan_file: load_model_gfpgan(req.use_face_correction)