diff --git a/invokeai/backend/model_management/model_probe.py b/invokeai/backend/model_management/model_probe.py index 21462cf6e6..1fcb5334df 100644 --- a/invokeai/backend/model_management/model_probe.py +++ b/invokeai/backend/model_management/model_probe.py @@ -345,6 +345,8 @@ class LoRACheckpointProbe(CheckpointProbeBase): return BaseModelType.StableDiffusion1 elif lora_token_vector_length == 1024: return BaseModelType.StableDiffusion2 + elif lora_token_vector_length is None: # variant w/o the text encoder! + return BaseModelType.StableDiffusion1 else: raise InvalidModelException(f"Unknown LoRA type")