diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-05-31 16:15:21 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-31 16:15:21 +0000 |
| commit | 52b8752e6201e24c783f674f8dc0681027e10ea9 (patch) | |
| tree | 30470f64d84cadc4140023c0bf34c35bb68f79b0 /modules/api/api.py | |
| parent | 00dfe27f59727407c5b408a80ff2a262934df495 (diff) | |
| parent | 78a602ae8c006077ca93913576335a3a33dba7cb (diff) | |
| download | stable-diffusion-webui-gfx803-52b8752e6201e24c783f674f8dc0681027e10ea9.tar.gz stable-diffusion-webui-gfx803-52b8752e6201e24c783f674f8dc0681027e10ea9.tar.bz2 stable-diffusion-webui-gfx803-52b8752e6201e24c783f674f8dc0681027e10ea9.zip | |
Merge branch 'dev' into report-error
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 79ce9228..fbd616a3 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -24,6 +24,7 @@ from modules.textual_inversion.preprocess import preprocess from modules.hypernetworks.hypernetwork import create_hypernetwork, train_hypernetwork from PIL import PngImagePlugin,Image from modules.sd_models import checkpoints_list, unload_model_weights, reload_model_weights +from modules.sd_vae import vae_dict from modules.sd_models_config import find_checkpoint_config_near_filename from modules.realesrgan_model import get_realesrgan_models from modules import devices @@ -190,6 +191,7 @@ class Api: self.add_api_route("/sdapi/v1/samplers", self.get_samplers, methods=["GET"], response_model=List[models.SamplerItem]) self.add_api_route("/sdapi/v1/upscalers", self.get_upscalers, methods=["GET"], response_model=List[models.UpscalerItem]) self.add_api_route("/sdapi/v1/sd-models", self.get_sd_models, methods=["GET"], response_model=List[models.SDModelItem]) + self.add_api_route("/sdapi/v1/sd-vae", self.get_sd_vaes, methods=["GET"], response_model=List[models.SDVaeItem]) self.add_api_route("/sdapi/v1/hypernetworks", self.get_hypernetworks, methods=["GET"], response_model=List[models.HypernetworkItem]) self.add_api_route("/sdapi/v1/face-restorers", self.get_face_restorers, methods=["GET"], response_model=List[models.FaceRestorerItem]) self.add_api_route("/sdapi/v1/realesrgan-models", self.get_realesrgan_models, methods=["GET"], response_model=List[models.RealesrganItem]) @@ -542,6 +544,9 @@ class Api: def get_sd_models(self): return [{"title": x.title, "model_name": x.model_name, "hash": x.shorthash, "sha256": x.sha256, "filename": x.filename, "config": find_checkpoint_config_near_filename(x)} for x in checkpoints_list.values()] + def get_sd_vaes(self): + return [{"model_name": x, "filename": vae_dict[x]} for x in vae_dict.keys()] + def get_hypernetworks(self): return [{"name": name, "path": shared.hypernetworks[name]} for name in shared.hypernetworks] |