diff --git a/docker_images/diffusers/app/lora.py b/docker_images/diffusers/app/lora.py index 355dae78..3c508cd6 100644 --- a/docker_images/diffusers/app/lora.py +++ b/docker_images/diffusers/app/lora.py @@ -158,9 +158,10 @@ def _load_lora_adapter(self, kwargs): logger.error(msg) raise ValueError(msg) base_model = model_data.cardData["base_model"] - if ( - isinstance(base_model, list) and (self.model_id not in base_model) - ) or (self.model_id != base_model): + is_list = isinstance(base_model, list) + if (is_list and (self.model_id not in base_model)) or ( + not is_list and self.model_id != base_model + ): msg = f"Requested adapter {adapter:s} is not a LoRA adapter for base model {self.model_id:s}" logger.error(msg) raise ValueError(msg)