From 7269c9f02eaef194c1be613fd7e0e7ad666b143a Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Mon, 8 Jan 2024 17:18:26 -0500 Subject: [PATCH] Enable correct probing of LoRA latent-consistency/lcm-lora-sdxl (#5449) - Closes #5435 Co-authored-by: Lincoln Stein --- 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 d312c6a0b4..74b1b72d31 100644 --- a/invokeai/backend/model_management/model_probe.py +++ b/invokeai/backend/model_management/model_probe.py @@ -370,6 +370,8 @@ class LoRACheckpointProbe(CheckpointProbeBase): return BaseModelType.StableDiffusion1 elif token_vector_length == 1024: return BaseModelType.StableDiffusion2 + elif token_vector_length == 1280: + return BaseModelType.StableDiffusionXL # recognizes format at https://civitai.com/models/224641 elif token_vector_length == 2048: return BaseModelType.StableDiffusionXL else: