api endpoint refresh vaes

This commit is contained in:
Vladimir Mandic
2023-07-26 15:51:19 -04:00
parent 15d5873f22
commit ff9b8bc062
5 changed files with 25 additions and 3 deletions
+5 -1
View File
@@ -132,6 +132,7 @@ class Api:
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-checkpoints", self.refresh_checkpoints, methods=["POST"])
self.add_api_route("/sdapi/v1/refresh-vaes", self.refresh_vaes, methods=["POST"])
self.add_api_route("/sdapi/v1/create/embedding", self.create_embedding, methods=["POST"], response_model=models.CreateResponse)
self.add_api_route("/sdapi/v1/create/hypernetwork", self.create_hypernetwork, methods=["POST"], response_model=models.CreateResponse)
self.add_api_route("/sdapi/v1/preprocess", self.preprocess, methods=["POST"], response_model=models.PreprocessResponse)
@@ -498,7 +499,10 @@ class Api:
}
def refresh_checkpoints(self):
shared.refresh_checkpoints()
return shared.refresh_checkpoints()
def refresh_vaes(self):
return shared.refresh_vaes()
def create_embedding(self, args: dict):
try: