From 477b5260aafb5bd9502c7ddddf25665d83eed6b6 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sun, 24 Sep 2023 14:47:13 +0900 Subject: [PATCH] fix sai metadata for sdxl closes #824 --- networks/extract_lora_from_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/networks/extract_lora_from_models.py b/networks/extract_lora_from_models.py index eed30350..dba7cd4e 100644 --- a/networks/extract_lora_from_models.py +++ b/networks/extract_lora_from_models.py @@ -14,7 +14,7 @@ import lora CLAMP_QUANTILE = 0.99 -MIN_DIFF = 1e-4 +MIN_DIFF = 1e-1 def save_to_file(file_name, model, state_dict, dtype): @@ -200,7 +200,7 @@ def svd(args): if not args.no_metadata: title = os.path.splitext(os.path.basename(args.save_to))[0] sai_metadata = sai_model_spec.build_metadata( - None, args.v2, args.v_parameterization, False, True, False, time.time(), title=title + None, args.v2, args.v_parameterization, args.sdxl, True, False, time.time(), title=title ) metadata.update(sai_metadata)