From de2879f60250a605a079eb5526ea59c4b6ee659a Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sun, 10 Dec 2023 15:48:02 -0500 Subject: [PATCH] port new code for detecting sdxl-based embeddings --- invokeai/backend/model_manager/probe.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/invokeai/backend/model_manager/probe.py b/invokeai/backend/model_manager/probe.py index 7cade3cada..1fda6e376b 100644 --- a/invokeai/backend/model_manager/probe.py +++ b/invokeai/backend/model_manager/probe.py @@ -416,6 +416,8 @@ class TextualInversionCheckpointProbe(CheckpointProbeBase): token_dim = list(checkpoint["string_to_param"].values())[0].shape[-1] elif "emb_params" in checkpoint: token_dim = checkpoint["emb_params"].shape[-1] + elif "clip_g" in checkpoint: + token_dim = checkpoint["clip_g"].shape[-1] else: token_dim = list(checkpoint.values())[0].shape[0] if token_dim == 768: