From f3d3316558d29bed477776876c940fe52b1118ef Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sun, 6 Aug 2023 16:00:53 -0400 Subject: [PATCH] probe LoRAs that do not have the text encoder --- invokeai/backend/model_management/model_probe.py | 2 ++ 1 file changed, 2 insertions(+) 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")