add queue lock for refresh-checkpoints

089a0022 · tangjicheng · 394ffa7b · 089a0022
Commit 089a0022 authored Jul 10, 2023 by tangjicheng
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

modules/api/api.py modules/api/api.py +2 -1

No files found.
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -593,7 +593,8 @@ class Api:
        }

    def refresh_checkpoints(self):
-        shared.refresh_checkpoints()
+        with self.queue_lock:
+            shared.refresh_checkpoints()

    def create_embedding(self, args: dict):
        try: