Merge pull request #10845 from DragonHawkAlpha/master (69f49a93) · Commits · github_fork / Stable Diffusion Webui

modules/api/api.py

+5 −0

Original line number	Diff line number	Diff line
		@@ -23,6 +23,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
		@@ -189,6 +190,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])
		@@ -541,6 +543,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]

modules/api/models.py

+4 −0

Original line number	Diff line number	Diff line
		@@ -249,6 +249,10 @@ class SDModelItem(BaseModel):
		filename: str = Field(title="Filename")
		config: Optional[str] = Field(title="Config file")

		class SDVaeItem(BaseModel):
		model_name: str = Field(title="Model Name")
		filename: str = Field(title="Filename")

		class HypernetworkItem(BaseModel):
		name: str = Field(title="Name")
		path: Optional[str] = Field(title="Path")

Admin message