From 8500bac3ca3e6ad4d90a7380264bf307fe80b0ab Mon Sep 17 00:00:00 2001 From: Sergey Borisov Date: Sun, 28 Jul 2024 22:51:52 +0300 Subject: [PATCH] Use logger for warning --- invokeai/backend/lora.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/invokeai/backend/lora.py b/invokeai/backend/lora.py index f5f3eedaa8..b4b59b5fcf 100644 --- a/invokeai/backend/lora.py +++ b/invokeai/backend/lora.py @@ -9,6 +9,7 @@ import torch from safetensors.torch import load_file from typing_extensions import Self +import invokeai.backend.util.logging as logger from invokeai.backend.model_manager import BaseModelType from invokeai.backend.raw_model import RawModel @@ -77,9 +78,8 @@ class LoRALayerBase: all_known_keys = known_keys | {"alpha", "bias_indices", "bias_values", "bias_size"} unknown_keys = set(values.keys()) - all_known_keys if unknown_keys: - # TODO: how to warn log? - print( - f"[WARN] Unexpected keys found in LoRA/LyCORIS layer, model might work incorrectly! Keys: {unknown_keys}" + logger.warning( + f"Unexpected keys found in LoRA/LyCORIS layer, model might work incorrectly! Keys: {unknown_keys}" )