Merge branch 'AUTOMATIC1111:master' into master

1 files changed, 6 insertions, 2 deletions

diff --git a/modules/cache.py b/modules/cache.py
index 71fe6302..ff26a213 100644
--- a/modules/cache.py
+++ b/modules/cache.py
@@ -1,11 +1,12 @@
 import json

+import os

 import os.path

 import threading

 import time

 

 from modules.paths import data_path, script_path

 

-cache_filename = os.path.join(data_path, "cache.json")

+cache_filename = os.environ.get('SD_WEBUI_CACHE_FILE', os.path.join(data_path, "cache.json"))

 cache_data = None

 cache_lock = threading.Lock()

 

@@ -29,9 +30,12 @@ def dump_cache():
             time.sleep(1)

 

         with cache_lock:

-            with open(cache_filename, "w", encoding="utf8") as file:

+            cache_filename_tmp = cache_filename + "-"

+            with open(cache_filename_tmp, "w", encoding="utf8") as file:

                 json.dump(cache_data, file, indent=4)

 

+            os.replace(cache_filename_tmp, cache_filename)

+

             dump_cache_after = None

             dump_cache_thread = None