From c9a5e729d91ba3772d421a5db7bf79497011e59f Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Mon, 13 Nov 2023 22:57:35 -0800 Subject: [PATCH] fix #938 --- fooocus_version.py | 2 +- modules/lora.py | 14 ++++++++++---- update_log.md | 4 ++++ 3 files changed, 15 insertions(+), 5 deletions(-) diff --git a/fooocus_version.py b/fooocus_version.py index de300a02..36f0485f 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.1.805' +version = '2.1.806' diff --git a/modules/lora.py b/modules/lora.py index 41c319d0..832750de 100644 --- a/modules/lora.py +++ b/modules/lora.py @@ -136,7 +136,13 @@ def load_dangerous_lora(lora, to_load): patch_dict["{}.bias".format(to_load[x][:-len(".weight")])] = (diff_bias,) loaded_keys.add(diff_bias_name) - for x in lora.keys(): - if x not in loaded_keys: - return {} - return patch_dict + remaining_keys = [x for x in lora.keys() if x not in loaded_keys] + + if len(remaining_keys) == 0: + return patch_dict + + if len(remaining_keys) < 8: + print(f'LoRA loaded with extra keys: {remaining_keys}') + return patch_dict + + return {} diff --git a/update_log.md b/update_log.md index 0efa5ac5..89b05713 100644 --- a/update_log.md +++ b/update_log.md @@ -1,3 +1,7 @@ +# 2.1.806 + +* fix some lora problems related to clip. + # 2.1.805 * Responsive UI for small screens.