From 37de18419823b789363c30761d1a615af70a534b Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Tue, 23 Sep 2025 14:31:04 +1000 Subject: [PATCH] fix(mm): tis use existing weight_files method --- invokeai/backend/model_manager/config.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/invokeai/backend/model_manager/config.py b/invokeai/backend/model_manager/config.py index 2d5db0fdda..7b0bcea8e5 100644 --- a/invokeai/backend/model_manager/config.py +++ b/invokeai/backend/model_manager/config.py @@ -607,7 +607,6 @@ class ControlNetCheckpointConfig(CheckpointConfigBase, ControlAdapterConfigBase, class TextualInversionConfigBase(ABC, BaseModel): type: Literal[ModelType.TextualInversion] = ModelType.TextualInversion - KNOWN_SUFFIXES: ClassVar = {"bin", "safetensors", "pt", "ckpt"} KNOWN_KEYS: ClassVar = {"string_to_param", "emb_params", "clip_g"} @classmethod @@ -621,7 +620,7 @@ class TextualInversionConfigBase(ABC, BaseModel): if p.is_dir(): return False - if p.name in [f"learned_embeds.{s}" for s in cls.KNOWN_SUFFIXES]: + if p.name in [f"learned_embeds.{s}" for s in mod.weight_files()]: return True state_dict = mod.load_state_dict(p)