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

Merge pull request #300 from cmdr2/beta

Browse Source 
Don't delete the model variables twice
This commit is contained in:
cmdr2 2022-10-08 13:59:06 +05:30 committed by GitHub
commit ad617e0deb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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' device = 'cpu'
if model_is_half: if model_is_half:
del model, modelCS, modelFS
load_model_ckpt(ckpt_to_use, device) load_model_ckpt(ckpt_to_use, device)
needs_model_reload = False 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 \ 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): (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')) load_model_ckpt(ckpt_to_use, device, req.turbo, unet_bs, ('full' if req.use_full_precision else 'autocast'))
needs_model_reload = False needs_model_reload = False