TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Attempt to extract bucket_info from ss_datasets.

Browse Source
This commit is contained in:
Christian Bastian
2024-07-19 03:29:32 -04:00
parent 4966348d55
commit 7be410dabc
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
__init__.py
View File

@@ -810,9 +810,16 @@ async def get_model_info(request):
notes = f.read() notes = f.read()
if metadata is not None: if metadata is not None:
img_buckets = metadata.get("ss_bucket_info", "{}") img_buckets = metadata.get("ss_bucket_info", None)
datasets = metadata.get("ss_datasets", None)
if type(img_buckets) is str: if type(img_buckets) is str:
img_buckets = json.loads(img_buckets) img_buckets = json.loads(img_buckets)
elif type(datasets) is str:
datasets = json.loads(datasets)
if isinstance(datasets, list):
datasets = datasets[0]
img_buckets = datasets.get("bucket_info", None)
resolutions = {} resolutions = {}
if img_buckets is not None: if img_buckets is not None:
buckets = img_buckets.get("buckets", {}) buckets = img_buckets.get("buckets", {})