From 1b4dbf5266ddfb2c96f4ef7b15b50cf86ffdd56e Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Tue, 7 Oct 2025 17:49:51 +1100 Subject: [PATCH] fix(mm): t5 identification --- .../model_manager/configs/t5_encoder.py | 19 ++++++------------- 1 file changed, 6 insertions(+), 13 deletions(-) diff --git a/invokeai/backend/model_manager/configs/t5_encoder.py b/invokeai/backend/model_manager/configs/t5_encoder.py index 3b6f2e733d..ed682e1430 100644 --- a/invokeai/backend/model_manager/configs/t5_encoder.py +++ b/invokeai/backend/model_manager/configs/t5_encoder.py @@ -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)