docs(mm): add todos

Author: psychedelicious

invokeai/backend/model_manager/config.py

@@ -1133,9 +1133,11 @@ def from_model_on_disk(cls, mod: ModelOnDisk, fields: dict[str, Any]) -> Self:
         }:
             variant = fields.get("variant") or cls._get_variant_or_raise(mod, base)
             prediction_type = fields.get("prediction_type") or cls._get_scheduler_prediction_type_or_raise(mod, base)
-            upcast_attention = fields.get("upcast_attention") or cls._get_upcast_attention_or_raise(base, prediction_type)
+            upcast_attention = fields.get("upcast_attention") or cls._get_upcast_attention_or_raise(
+                base, prediction_type
+            )
         else:
-            variant= None
+            variant = None
             prediction_type = None
             upcast_attention = False
 
@@ -1144,7 +1146,16 @@ def from_model_on_disk(cls, mod: ModelOnDisk, fields: dict[str, Any]) -> Self:
         else:
             submodels = None
 
-        return cls(**fields, base=base, variant=variant, prediction_type=prediction_type, upcast_attention=upcast_attention, submodels=submodels,)
+        return cls(
+            **fields,
+            base=base,
+            # TODO(psyche): figure out variant/prediction_type/upcast_attention
+            variant=variant,
+            prediction_type=prediction_type,
+            upcast_attention=upcast_attention,
+            # TODO(psyche): This is only for SD3 models - split up the config classes
+            submodels=submodels,
+        )
 
     @classmethod
     def _get_base_or_raise(cls, mod: ModelOnDisk) -> MainDiffusers_SupportedBases:
@@ -1282,9 +1293,10 @@ def _get_submodels_or_raise(
 
         return submodels
 
-
     @classmethod
-    def _get_upcast_attention_or_raise(cls, base: MainDiffusers_SupportedBases, prediction_type: SchedulerPredictionType) -> bool:
+    def _get_upcast_attention_or_raise(
+        cls, base: MainDiffusers_SupportedBases, prediction_type: SchedulerPredictionType
+    ) -> bool:
         if base not in {
             BaseModelType.StableDiffusion1,
             BaseModelType.StableDiffusion2,
@@ -1298,6 +1310,7 @@ def _get_upcast_attention_or_raise(cls, base: MainDiffusers_SupportedBases, pred
 
         return False
 
+
 class IPAdapterConfigBase(ABC, BaseModel):
     type: Literal[ModelType.IPAdapter] = Field(default=ModelType.IPAdapter)