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

fix(mm): t5 identification

Browse Source
This commit is contained in:
psychedelicious
2025-10-07 17:49:51 +11:00
parent 56e31ca4ac
commit 2e5ec1c98b
1 changed files with 6 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
invokeai/backend/model_manager/configs/t5_encoder.py
View File

@@ -5,7 +5,6 @@ from pydantic import Field
from invokeai.backend.model_manager.configs.base import Config_Base
from invokeai.backend.model_manager.configs.identification_utils import (
NotAMatchError,
common_config_paths,
raise_for_class_name,
raise_for_override_fields,
raise_if_not_dir,
@@ -29,12 +28,9 @@ class T5Encoder_T5Encoder_Config(Config_Base):
raise_for_override_fields(cls, override_fields)
raise_for_class_name(
common_config_paths(mod.path),
{
"T5EncoderModel",
},
)
expected_config_path = mod.path / "text_encoder_2" / "config.json"
expected_class_name = "T5EncoderModel"
raise_for_class_name(expected_config_path, expected_class_name)
cls.raise_if_doesnt_have_unquantized_config_file(mod)
@@ -61,12 +57,9 @@ class T5Encoder_BnBLLMint8_Config(Config_Base):
raise_for_override_fields(cls, override_fields)
raise_for_class_name(
common_config_paths(mod.path),
{
"T5EncoderModel",
},
)
expected_config_path = mod.path / "text_encoder_2" / "config.json"
expected_class_name = "T5EncoderModel"
raise_for_class_name(expected_config_path, expected_class_name)
cls.raise_if_filename_doesnt_look_like_bnb_quantized(mod)