From 07a667ad9f115126f7e74acaac5361b09f207566 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Tue, 7 Oct 2025 15:22:36 +1100
Subject: [PATCH] feat(mm): add helper method for legacy configs

---
 invokeai/backend/model_manager/single_file_config_files.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/invokeai/backend/model_manager/single_file_config_files.py b/invokeai/backend/model_manager/single_file_config_files.py
index 22fe646b55..fa4b9e934b 100644
--- a/invokeai/backend/model_manager/single_file_config_files.py
+++ b/invokeai/backend/model_manager/single_file_config_files.py
@@ -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(