aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-01-04 10:26:23 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-01-04 10:26:23 +0000
commit4ec6470a1a2d9430b91266426f995e48f59564e1 (patch)
treeedcb2d5e124718592450a7d13d6369c41bbe6f62
parent8d8a05a3bbb50fdfeab51679a919d2487bd97976 (diff)
downloadstable-diffusion-webui-gfx803-4ec6470a1a2d9430b91266426f995e48f59564e1.tar.gz
stable-diffusion-webui-gfx803-4ec6470a1a2d9430b91266426f995e48f59564e1.tar.bz2
stable-diffusion-webui-gfx803-4ec6470a1a2d9430b91266426f995e48f59564e1.zip
fix checkpoint list API
-rw-r--r--modules/api/api.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/api/api.py b/modules/api/api.py
index 9c670f00..2b1f180c 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -18,7 +18,7 @@ from modules.textual_inversion.textual_inversion import create_embedding, train_
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
+from modules.sd_models import checkpoints_list, find_checkpoint_config
from modules.realesrgan_model import get_realesrgan_models
from modules import devices
from typing import List
@@ -303,7 +303,7 @@ class Api:
return upscalers
def get_sd_models(self):
- return [{"title":x.title, "model_name":x.model_name, "hash":x.hash, "filename": x.filename, "config": x.config} for x in checkpoints_list.values()]
+ return [{"title":x.title, "model_name":x.model_name, "hash":x.hash, "filename": x.filename, "config": find_checkpoint_config(x)} for x in checkpoints_list.values()]
def get_hypernetworks(self):
return [{"name": name, "path": shared.hypernetworks[name]} for name in shared.hypernetworks]