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

feat(mm): add helper method for legacy configs

Browse Source
This commit is contained in:
psychedelicious
2025-10-07 15:22:36 +11:00
parent fcf0200168
commit 3a76d26776
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/backend/model_manager/single_file_config_files.py
View File

@@ -1,5 +1,6 @@
from dataclasses import dataclass
from invokeai.backend.model_manager.configs.factory import AnyModelConfig
from invokeai.backend.model_manager.taxonomy import (
BaseModelType,
ModelType,
@@ -15,6 +16,12 @@ class LegacyConfigKey:
variant: ModelVariantType | None = None
pred: SchedulerPredictionType | None = None
@classmethod
def from_model_config(cls, config: AnyModelConfig) -> "LegacyConfigKey":
variant = getattr(config, "variant", None)
pred = getattr(config, "prediction_type", None)
return cls(type=config.type, base=config.base, variant=variant, pred=pred)
LEGACY_CONFIG_MAP: dict[LegacyConfigKey, str] = {
LegacyConfigKey(