extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-27 09:28:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Don't delete the model variables twice

Browse Source
This commit is contained in:
cmdr2 2022-10-08 13:58:44 +05:30
parent 883dc72fc6
commit 7ae70d5a4d
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ui/sd_internal/runtime.py
View File

@ -249,7 +249,6 @@ def do_mk_img(req: Request):
device = 'cpu'
if model_is_half:
del model, modelCS, modelFS
load_model_ckpt(ckpt_to_use, device)
needs_model_reload = False
@ -263,7 +262,6 @@ def do_mk_img(req: Request):
if (precision == 'autocast' and (req.use_full_precision or not model_is_half)) or \
(precision == 'full' and not req.use_full_precision and not force_full_precision):
del model, modelCS, modelFS
load_model_ckpt(ckpt_to_use, device, req.turbo, unet_bs, ('full' if req.use_full_precision else 'autocast'))
needs_model_reload = False