mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-02-19 03:49:58 +00:00
Actually fixed meta data
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user