From 1f91b35a43858d8884d2b08d725aa46ec67a0f6e Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Sat, 31 Aug 2024 10:20:22 -0700
Subject: [PATCH] add signal_empty_cache

---
 backend/memory_management.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/backend/memory_management.py b/backend/memory_management.py
index cdcbbed6..eddf5224 100644
--- a/backend/memory_management.py
+++ b/backend/memory_management.py
@@ -346,6 +346,9 @@ def bake_gguf_model(model):
         if gguf_cls is not None:
             gguf_cls.bake(p)
 
+    global signal_empty_cache
+    signal_empty_cache = True
+
     model.gguf_baked = True
     return model