diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-23 22:35:41 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-23 22:35:41 +0300 |
| commit | 19c95de8eb6a1226f0cd1aaeaa9768cf3cbfdef9 (patch) | |
| tree | 06b9104bdefa0b41a3d3eb6eb32b5a079f17ea38 /modules | |
| parent | 358e9e2847efae0cd45107edbf050925a1a068df (diff) | |
| parent | 2974b9cee94dc474ffbc9e9617d14c9aaf9e1e63 (diff) | |
Merge pull request #14715 from stefanbenten/sb/embedding-refresh
modules/api/api.py: add api endpoint to refresh embeddings list
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/api/api.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index 59e46335..4e656082 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -230,6 +230,7 @@ class Api: self.add_api_route("/sdapi/v1/realesrgan-models", self.get_realesrgan_models, methods=["GET"], response_model=list[models.RealesrganItem]) self.add_api_route("/sdapi/v1/prompt-styles", self.get_prompt_styles, methods=["GET"], response_model=list[models.PromptStyleItem]) self.add_api_route("/sdapi/v1/embeddings", self.get_embeddings, methods=["GET"], response_model=models.EmbeddingsResponse) + self.add_api_route("/sdapi/v1/refresh-embeddings", self.refresh_embeddings, methods=["POST"]) self.add_api_route("/sdapi/v1/refresh-checkpoints", self.refresh_checkpoints, methods=["POST"]) self.add_api_route("/sdapi/v1/refresh-vae", self.refresh_vae, methods=["POST"]) self.add_api_route("/sdapi/v1/create/embedding", self.create_embedding, methods=["POST"], response_model=models.CreateResponse) @@ -747,6 +748,10 @@ class Api: "skipped": convert_embeddings(db.skipped_embeddings), } + def refresh_embeddings(self): + with self.queue_lock: + sd_hijack.model_hijack.embedding_db.load_textual_inversion_embeddings(force_reload=True) + def refresh_checkpoints(self): with self.queue_lock: shared.refresh_checkpoints() |