From c03a1918e7d37db4187794d4160cd9062db5c500 Mon Sep 17 00:00:00 2001 From: Qing Date: Sat, 10 Feb 2024 16:41:54 +0800 Subject: [PATCH] only auto switch to lcm schduler when using lcm lora --- iopaint/model/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/iopaint/model/base.py b/iopaint/model/base.py index 2f23d70..f8efd8e 100644 --- a/iopaint/model/base.py +++ b/iopaint/model/base.py @@ -395,7 +395,7 @@ class DiffusionInpaintModel(InpaintModel): def set_scheduler(self, config: InpaintRequest): scheduler_config = self.model.scheduler.config sd_sampler = config.sd_sampler - if config.sd_lcm_lora: + if config.sd_lcm_lora and self.model_info.support_lcm_lora: sd_sampler = SDSampler.lcm logger.info(f"LCM Lora enabled, use {sd_sampler} sampler") scheduler = get_scheduler(sd_sampler, scheduler_config)