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 probe to install Invoke formatted clip embeds and t5 encoders

Browse Source
This commit is contained in:
Brandon Rising 2024-08-29 13:04:24 -04:00
parent f2a1a39b33
commit be76109f3b
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/backend/model_manager/probe.py
View File

@ -304,6 +304,11 @@ class ModelProbe(object):
else:
error_hint = f"class {class_name} is not one of the supported classes [{', '.join(cls.CLASS2TYPE.keys())}]"
if (folder_path / "tokenizer").exists() and (folder_path / "text_encoder").exists():
return ModelType.CLIPEmbed
if (folder_path / "tokenizer_2").exists() and (folder_path / "text_encoder_2").exists():
return ModelType.T5Encoder
# give up
raise InvalidModelConfigException(
f"Unable to determine model type for {folder_path}" + (f"; {error_hint}" if error_hint else "")
@ -747,6 +752,10 @@ class TextualInversionFolderProbe(FolderProbeBase):
class T5EncoderFolderProbe(FolderProbeBase):
def get_base_type(self) -> BaseModelType:
return BaseModelType.Any
def get_format(self) -> ModelFormat:
return ModelFormat.T5Encoder