Empty cache after model unloading for normal vram and lower.

im-stuff · Jul 9, 2023 · 0ae81c0 · 0ae81c0
1 parent d3f5998
commit 0ae81c0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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: