Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update model_probe to work with diffuser-format SD TI embeddings.

Browse Source
This commit is contained in:
Ryan Dick 2023-12-15 17:56:49 -05:00 committed by Kent Keirsey
parent 0e738c4290
commit cb698ff1fb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/model_probe.py
View File

@ -389,7 +389,7 @@ class TextualInversionCheckpointProbe(CheckpointProbeBase):
elif "clip_g" in checkpoint: elif "clip_g" in checkpoint:
token_dim = checkpoint["clip_g"].shape[-1] token_dim = checkpoint["clip_g"].shape[-1]
else: else:
token_dim = list(checkpoint.values())[0].shape[0] token_dim = list(checkpoint.values())[0].shape[-1]
if token_dim == 768: if token_dim == 768:
return BaseModelType.StableDiffusion1 return BaseModelType.StableDiffusion1
elif token_dim == 1024: elif token_dim == 1024: