diff options
| author | ramyma <ramy.mahmoudi@gmail.com> | 2023-06-04 13:59:23 +0000 |
|---|---|---|
| committer | ramyma <ramy.mahmoudi@gmail.com> | 2023-06-04 14:05:29 +0000 |
| commit | 4faaf3e723ddaec022af85d6e66c3b1bac449584 (patch) | |
| tree | 103d371fb984ba553374ad70a801867c8ab035fe | |
| parent | fbf88343deff2f0d6c1c375ec858c094ed9fa260 (diff) | |
| download | stable-diffusion-webui-gfx803-4faaf3e723ddaec022af85d6e66c3b1bac449584.tar.gz stable-diffusion-webui-gfx803-4faaf3e723ddaec022af85d6e66c3b1bac449584.tar.bz2 stable-diffusion-webui-gfx803-4faaf3e723ddaec022af85d6e66c3b1bac449584.zip | |
Add endpoint to get latent_upscale_modes for hires fix
| -rw-r--r-- | modules/api/api.py | 9 | ||||
| -rw-r--r-- | modules/api/models.py | 3 |
2 files changed, 12 insertions, 0 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index d34ab422..61a47005 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -189,6 +189,7 @@ class Api: self.add_api_route("/sdapi/v1/cmd-flags", self.get_cmd_flags, methods=["GET"], response_model=models.FlagsModel) 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/latent-upscale-modes", self.get_latent_upscale_modes, methods=["GET"], response_model=List[models.LatentUpscalerModeItem]) 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]) @@ -540,6 +541,14 @@ class Api: for upscaler in shared.sd_upscalers ] + def get_latent_upscale_modes(self): + return [ + { + "name": upscale_mode, + } + for upscale_mode in [*(shared.latent_upscale_modes or {})] + ] + 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()] diff --git a/modules/api/models.py b/modules/api/models.py index 47fdede2..b3a745f0 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -241,6 +241,9 @@ class UpscalerItem(BaseModel): model_url: Optional[str] = Field(title="URL") scale: Optional[float] = Field(title="Scale") +class LatentUpscalerModeItem(BaseModel): + name: str = Field(title="Name") + class SDModelItem(BaseModel): title: str = Field(title="Title") model_name: str = Field(title="Model Name") |