Attempt to extract bucket_info from ss_datasets.

This commit is contained in:
Christian Bastian
2024-07-19 03:29:32 -04:00
parent 4966348d55
commit 7be410dabc

View File

@@ -810,9 +810,16 @@ async def get_model_info(request):
notes = f.read()
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:
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 = {}
if img_buckets is not None:
buckets = img_buckets.get("buckets", {})