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

fix prints

Browse Source
This commit is contained in:
layerdiffusion
2024-08-21 22:24:54 -07:00
parent c1614e08c6
commit 909ad6c734
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/memory_management.py
View File

@@ -469,8 +469,12 @@ def unload_model_clones(model):
current_loaded_models.pop(i).model_unload(avoid_model_moving=True)
def free_memory(memory_required, device, keep_loaded=[]):
print(f"[Unload] Trying to free {memory_required / (1024 * 1024):.2f} MB for {device} with {len(keep_loaded)} models keep loaded ...")
def free_memory(memory_required, device, keep_loaded=[], free_all=False):
if free_all:
memory_required = 1e30
print(f"[Unload] Trying to free all memory for {device} with {len(keep_loaded)} models keep loaded ...")
else:
print(f"[Unload] Trying to free {memory_required / (1024 * 1024):.2f} MB for {device} with {len(keep_loaded)} models keep loaded ...")
offload_everything = ALWAYS_VRAM_OFFLOAD or vram_state == VRAMState.NO_VRAM
unloaded_model = False
@@ -1100,4 +1104,4 @@ def soft_empty_cache(force=False):
def unload_all_models():
free_memory(1e30, get_torch_device())
free_memory(1e30, get_torch_device(), free_all=True)