From f197a37c12ded65dd58ebc7a44a94f81b4d744ef Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 15 Oct 2024 08:58:26 +0530 Subject: [PATCH] fix --- src/diffusers/loaders/lora_pipeline.py | 6 +++--- src/diffusers/loaders/unet.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/diffusers/loaders/lora_pipeline.py b/src/diffusers/loaders/lora_pipeline.py index dc42e5c67770..5e01ec567f9a 100644 --- a/src/diffusers/loaders/lora_pipeline.py +++ b/src/diffusers/loaders/lora_pipeline.py @@ -1956,7 +1956,7 @@ def load_lora_into_transformer( lora_unexpected_keys = [k for k in unexpected_keys if "lora_" in k and adapter_name in k] if lora_unexpected_keys: warn_msg = ( - f"Loading adapter weights from state_dict led to unexpected keys not found in the model:" + f"Loading adapter weights from state_dict led to unexpected keys found in the model:" f" {', '.join(lora_unexpected_keys)}. " ) @@ -2319,7 +2319,7 @@ def load_lora_into_transformer(cls, state_dict, network_alphas, transformer, ada lora_unexpected_keys = [k for k in unexpected_keys if "lora_" in k and adapter_name in k] if lora_unexpected_keys: warn_msg = ( - f"Loading adapter weights from state_dict led to unexpected keys not found in the model:" + f"Loading adapter weights from state_dict led to unexpected keys found in the model:" f" {', '.join(lora_unexpected_keys)}. " ) @@ -2773,7 +2773,7 @@ def load_lora_into_transformer( lora_unexpected_keys = [k for k in unexpected_keys if "lora_" in k and adapter_name in k] if lora_unexpected_keys: warn_msg = ( - f"Loading adapter weights from state_dict led to unexpected keys not found in the model:" + f"Loading adapter weights from state_dict led to unexpected keys found in the model:" f" {', '.join(lora_unexpected_keys)}. " ) diff --git a/src/diffusers/loaders/unet.py b/src/diffusers/loaders/unet.py index 8f302bdad31b..2fa7732a6a3b 100644 --- a/src/diffusers/loaders/unet.py +++ b/src/diffusers/loaders/unet.py @@ -362,7 +362,7 @@ def _process_lora( lora_unexpected_keys = [k for k in unexpected_keys if "lora_" in k and adapter_name in k] if lora_unexpected_keys: warn_msg = ( - f"Loading adapter weights from state_dict led to unexpected keys not found in the model:" + f"Loading adapter weights from state_dict led to unexpected keys found in the model:" f" {', '.join(lora_unexpected_keys)}. " )