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): tis use existing weight_files method

Browse Source
This commit is contained in:
psychedelicious
2025-09-23 14:31:04 +10:00
parent 4ae20f4876
commit 73b6fae00e
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/backend/model_manager/config.py
View File

@@ -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)