From 803b68d210515080a2901ed8a19aa66f805dbe8c Mon Sep 17 00:00:00 2001 From: Qing Date: Tue, 27 Feb 2024 22:42:33 +0800 Subject: [PATCH] fix lcm enable/disable bug --- iopaint/model_manager.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/iopaint/model_manager.py b/iopaint/model_manager.py index de13612..15cf32b 100644 --- a/iopaint/model_manager.py +++ b/iopaint/model_manager.py @@ -181,11 +181,16 @@ class ModelManager: lcm_lora_loaded = bool(self.model.model.get_list_adapters()) if config.sd_lcm_lora: if not lcm_lora_loaded: + logger.info("Load LCM LORA") self.model.model.load_lora_weights( self.model.lcm_lora_id, weight_name="pytorch_lora_weights.safetensors", local_files_only=is_local_files_only(), ) + else: + logger.info("Enable LCM LORA") + self.model.model.enable_lora() else: if lcm_lora_loaded: + logger.info("Disable LCM LORA") self.model.model.disable_lora()