From c3cbb6a48ba4d1019dc69801fd5fbe85c004a64e Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Tue, 12 Sep 2023 17:45:31 -0400 Subject: [PATCH] error handling on invalid metadata --- modules/sd_models.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 45bce3d0d..8cf20af56 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -362,11 +362,14 @@ def read_metadata_from_safetensors(filename): if large and k == 'ss_bucket_info': continue if v[0:1] == '{': - v = json.loads(v) - if large and k == 'ss_tag_frequency': - v = { i: len(j) for i, j in v.items() } - if large and k == 'sd_merge_models': - scrub_dict(v, ['sd_merge_recipe']) + try: + v = json.loads(v) + if large and k == 'ss_tag_frequency': + v = { i: len(j) for i, j in v.items() } + if large and k == 'sd_merge_models': + scrub_dict(v, ['sd_merge_recipe']) + except Exception: + pass res[k] = v sd_metadata[filename] = res global sd_metadata_pending # pylint: disable=global-statement