Skip to content

Commit cf0af13

Browse files
fix(mm): t5 identification
1 parent 468ae2f commit cf0af13

File tree

1 file changed

+6
-13
lines changed

1 file changed

+6
-13
lines changed

invokeai/backend/model_manager/configs/t5_encoder.py

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from invokeai.backend.model_manager.configs.base import Config_Base
66
from invokeai.backend.model_manager.configs.identification_utils import (
77
NotAMatchError,
8-
common_config_paths,
98
raise_for_class_name,
109
raise_for_override_fields,
1110
raise_if_not_dir,
@@ -29,12 +28,9 @@ def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -
2928

3029
raise_for_override_fields(cls, override_fields)
3130

32-
raise_for_class_name(
33-
common_config_paths(mod.path),
34-
{
35-
"T5EncoderModel",
36-
},
37-
)
31+
expected_config_path = mod.path / "text_encoder_2" / "config.json"
32+
expected_class_name = "T5EncoderModel"
33+
raise_for_class_name(expected_config_path, expected_class_name)
3834

3935
cls.raise_if_doesnt_have_unquantized_config_file(mod)
4036

@@ -61,12 +57,9 @@ def from_model_on_disk(cls, mod: ModelOnDisk, override_fields: dict[str, Any]) -
6157

6258
raise_for_override_fields(cls, override_fields)
6359

64-
raise_for_class_name(
65-
common_config_paths(mod.path),
66-
{
67-
"T5EncoderModel",
68-
},
69-
)
60+
expected_config_path = mod.path / "text_encoder_2" / "config.json"
61+
expected_class_name = "T5EncoderModel"
62+
raise_for_class_name(expected_config_path, expected_class_name)
7063

7164
cls.raise_if_filename_doesnt_look_like_bnb_quantized(mod)
7265

0 commit comments

Comments
 (0)