Skip to content

Commit 22918f2

Browse files
feat(mm): add helper method for legacy configs
1 parent b4350c0 commit 22918f2

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

invokeai/backend/model_manager/single_file_config_files.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from dataclasses import dataclass
22

3+
from invokeai.backend.model_manager.configs.factory import AnyModelConfig
34
from invokeai.backend.model_manager.taxonomy import (
45
BaseModelType,
56
ModelType,
@@ -15,6 +16,12 @@ class LegacyConfigKey:
1516
variant: ModelVariantType | None = None
1617
pred: SchedulerPredictionType | None = None
1718

19+
@classmethod
20+
def from_model_config(cls, config: AnyModelConfig) -> "LegacyConfigKey":
21+
variant = getattr(config, "variant", None)
22+
pred = getattr(config, "prediction_type", None)
23+
return cls(type=config.type, base=config.base, variant=variant, pred=pred)
24+
1825

1926
LEGACY_CONFIG_MAP: dict[LegacyConfigKey, str] = {
2027
LegacyConfigKey(

0 commit comments

Comments
 (0)