--disable-smart-memory now unloads everything like it did originally.

2023-12-23 04:25:06 -05:00
parent 36a7953142
commit a252963f95
2 changed files with 6 additions and 0 deletions
--- a/execution.py
+++ b/execution.py
@@ -382,6 +382,8 @@ class PromptExecutor:
            for x in executed:
                self.old_prompt[x] = copy.deepcopy(prompt[x])
            self.server.last_node_id = None
+            if comfy.model_management.DISABLE_SMART_MEMORY:
+                comfy.model_management.unload_all_models()