From f23ee63cb340504c15607cc68798cd28abb88037 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Fri, 23 Aug 2024 08:56:57 -0700 Subject: [PATCH] always set empty cache signal as long as any patch happens --- backend/patcher/lora.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/backend/patcher/lora.py b/backend/patcher/lora.py index 9c8348f2..83da8fe5 100644 --- a/backend/patcher/lora.py +++ b/backend/patcher/lora.py @@ -346,6 +346,8 @@ class LoraLoader: # Patch + memory_management.signal_empty_cache = True + for key, current_patches in (tqdm(self.patches.items(), desc=f'Patching LoRAs for {type(self.model).__name__}') if len(self.patches) > 0 else self.patches): try: parent_layer, child_key, weight = utils.get_attr_with_parent(self.model, key)