Empty cache after model unloading for normal vram and lower.

1 year ago · 0ae81c03bb
1 changed files with 2 additions and 0 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@ -238,6 +238,8 @@ def unload_model():
        current_loaded_model.model_patches_to(current_loaded_model.offload_device)
        current_loaded_model.unpatch_model()
        current_loaded_model = None
+        if vram_state != VRAMState.HIGH_VRAM:
+            soft_empty_cache()

    if vram_state != VRAMState.HIGH_VRAM:
        if len(current_gpu_controlnets) > 0: