holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #299 from cmdr2/beta

Browse Source 
Unload the previous model while reloading
This commit is contained in:
cmdr2 2022-10-08 13:19:58 +05:30 committed by GitHub
commit 62d1a0291e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ui/sd_internal/runtime.py
View File

@ -86,6 +86,8 @@ def load_model_ckpt(ckpt_to_use, device_to_use='cuda', turbo=False, unet_bs_to_u
precision = precision_to_use if not force_full_precision else 'full' precision = precision_to_use if not force_full_precision else 'full'
unet_bs = unet_bs_to_use unet_bs = unet_bs_to_use
unload_model()
if device == 'cpu': if device == 'cpu':
precision = 'full' precision = 'full'
@ -140,6 +142,18 @@ def load_model_ckpt(ckpt_to_use, device_to_use='cuda', turbo=False, unet_bs_to_u
print('loaded ', ckpt_file, 'to', device, 'precision', precision) print('loaded ', ckpt_file, 'to', device, 'precision', precision)
def unload_model():
global model, modelCS, modelFS
if model is not None:
del model
del modelCS
del modelFS
model = None
modelCS = None
modelFS = None
def load_model_gfpgan(gfpgan_to_use): def load_model_gfpgan(gfpgan_to_use):
global gfpgan_file, model_gfpgan global gfpgan_file, model_gfpgan