github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

correct modelformat probe for t5 encoders

Browse Source
This commit is contained in:
Lincoln Stein
2024-08-31 20:24:09 -04:00
committed by Brandon
parent 3f7f9f8d61
commit b6434c5318
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -756,12 +756,18 @@ class TextualInversionFolderProbe(FolderProbeBase):
class T5EncoderFolderProbe(FolderProbeBase):
def get_format(self) -> ModelFormat:
return ModelFormat.T5Encoder
def get_base_type(self) -> BaseModelType:
return BaseModelType.Flux
def get_format(self) -> ModelFormat:
path = self.model_path / "text_encoder_2"
if (path / "model.safetensors.index.json").exists():
return ModelFormat.T5Encoder
files = path.glob("*.safetensors")
if any(x for x in files if "llm_int8" in x.as_posix()):
return ModelFormat.BnbQuantizedLlmInt8b
raise f"{self.model_path.as_posix()}: unknown model format"
class ONNXFolderProbe(PipelineFolderProbe):
def get_base_type(self) -> BaseModelType: