diff --git a/scripts/extract_locon.py b/scripts/extract_locon.py index 9a80d06f..33d9df65 100644 --- a/scripts/extract_locon.py +++ b/scripts/extract_locon.py @@ -134,7 +134,7 @@ def main(): save_meta = create_meta([ item_meta, extract_diff_meta - ]) + ], name=name) output_file_name = f"lyco_{name}_{process['mode']}_{linear_mode_param}_{conv_mode_param}.safetensors" output_path = os.path.join(output_folder, output_file_name) diff --git a/toolkit/metadata.py b/toolkit/metadata.py index 32ba7d23..d30acefb 100644 --- a/toolkit/metadata.py +++ b/toolkit/metadata.py @@ -27,6 +27,5 @@ def create_meta(dict_list, name=None): 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) + meta[key] = json.dumps(value) return meta