Fixed issue with meta.

This commit is contained in:
Jaret Burkett
2023-07-05 16:49:59 -06:00
parent e4de8983c9
commit 18077ba265
2 changed files with 10 additions and 3 deletions

View File

@@ -22,3 +22,11 @@ def create_meta(dict_list, name=None):
if name is not None:
meta_string = meta_string.replace("[name]", name)
return json.loads(meta_string)
def prep_meta_for_safetensors(meta):
# safetensors can only be one level deep
for key, value in meta.items():
if isinstance(value, dict):
meta[key] = json.dumps(value)
return meta