diff options
| author | Stefan Benten <mail@stefan-benten.de> | 2024-01-21 14:05:47 +0100 |
|---|---|---|
| committer | Stefan Benten <mail@stefan-benten.de> | 2024-01-21 14:05:47 +0100 |
| commit | 2974b9cee94dc474ffbc9e9617d14c9aaf9e1e63 (patch) | |
| tree | 7b1a31a824cc4d580d74449752999353b023002b /modules/api/api.py | |
| parent | cf2772fab0af5573da775e7437e6acdca424f26e (diff) | |
modules/api/api.py: add api endpoint to refresh embeddings list
Diffstat (limited to 'modules/api/api.py')
| -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 b3d74e51..b6bb9d06 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) @@ -643,6 +644,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() |