lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix for neveroom extension

Browse Source
This commit is contained in:
lllyasviel
2024-02-25 16:47:02 -08:00
parent a0c89fae12
commit 056d4d0f91
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ldm_patched/modules/model_management.py
View File

@@ -397,9 +397,10 @@ def unload_model_clones(model):
current_loaded_models.pop(i).model_unload(avoid_model_moving=True)
def free_memory(memory_required, device, keep_loaded=[]):
offload_everything = ALWAYS_VRAM_OFFLOAD or vram_state == VRAMState.NO_VRAM
unloaded_model = False
for i in range(len(current_loaded_models) -1, -1, -1):
if not ALWAYS_VRAM_OFFLOAD:
if not offload_everything:
if get_free_memory(device) > memory_required:
break
shift_model = current_loaded_models[i]