From daee4c0d8f77cd01c427b4202dd333d7f159fa19 Mon Sep 17 00:00:00 2001
From: hako-mikan <122196982+hako-mikan@users.noreply.github.com>
Date: Wed, 29 Jan 2025 06:04:44 +0900
Subject: [PATCH] Add force refresh to LoRA Loader refresh function (#2584)

---
 backend/patcher/lora.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/backend/patcher/lora.py b/backend/patcher/lora.py
index 42f379bb..57762424 100644
--- a/backend/patcher/lora.py
+++ b/backend/patcher/lora.py
@@ -299,10 +299,10 @@ class LoraLoader:
         self.loaded_hash = str([])
 
     @torch.inference_mode()
-    def refresh(self, lora_patches, offload_device=torch.device('cpu')):
+    def refresh(self, lora_patches, offload_device=torch.device('cpu'), force_refresh = False):
         hashes = str(list(lora_patches.keys()))
 
-        if hashes == self.loaded_hash:
+        if hashes == self.loaded_hash and not force_refresh:
             return
 
         # Merge Patches