From 8dd3a06945d5e2be6fab31d9ef709632f64fff32 Mon Sep 17 00:00:00 2001 From: Qing Date: Tue, 16 Jan 2024 22:10:26 +0800 Subject: [PATCH] fix lcm_lora offline --- iopaint/model_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/iopaint/model_manager.py b/iopaint/model_manager.py index 623fb84..6d4461b 100644 --- a/iopaint/model_manager.py +++ b/iopaint/model_manager.py @@ -182,7 +182,7 @@ class ModelManager: lcm_lora_loaded = bool(self.model.model.get_list_adapters()) if config.sd_lcm_lora: if not lcm_lora_loaded: - self.model.model.load_lora_weights(self.model.lcm_lora_id) + self.model.model.load_lora_weights(self.model.lcm_lora_id, weight_name="pytorch_lora_weights.safetensors") else: if lcm_lora_loaded: self.model.model.disable_lora()