add queue lock for load_model_weights

AIRI-AUBE · Mar 8, 2023 · 541d7c5 · 541d7c5
1 parent c25694c
commit 541d7c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/api/api.py b/modules/api/api.py
@@ -418,7 +418,8 @@ def invocations(self, req: InvocationsRequest):
                 response = requests.post(url=f'{api_endpoint}/sd/user', json=inputs)
                 if response.status_code == 200 and response.text != '':
                     shared.opts.data = json.loads(response.text)
-                    sd_models.reload_model_weights()
+                    with self.queue_lock:
+                        sd_models.reload_model_weights()
 
                 self.download_s3files(hypernetwork_s3uri, os.path.join(script_path, shared.cmd_opts.hypernetwork_dir))
                 hypernetworks.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)