lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-05-01 03:31:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_models.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 06:43:33 -08:00
parent 6b88c82733
commit 8e36af0f06
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/sd_models.py
View File

@@ -582,6 +582,9 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
timer = Timer() timer = Timer()
if model_data.sd_model: if model_data.sd_model:
if model_data.sd_model.filename == checkpoint_info.filename:
return model_data.sd_model
model_data.sd_model = None model_data.sd_model = None
devices.torch_gc() devices.torch_gc()
@@ -597,6 +600,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
checkpoints_loaded[checkpoint_info] = state_dict.copy() checkpoints_loaded[checkpoint_info] = state_dict.copy()
sd_model = forge_loader.load_model_for_a1111(timer=timer, checkpoint_info=checkpoint_info, state_dict=state_dict) sd_model = forge_loader.load_model_for_a1111(timer=timer, checkpoint_info=checkpoint_info, state_dict=state_dict)
sd_model.filename = checkpoint_info.filename
del state_dict del state_dict