diff --git a/invokeai/backend/model_manager/config.py b/invokeai/backend/model_manager/config.py
index 2937eb3a27..e95f851a62 100644
--- a/invokeai/backend/model_manager/config.py
+++ b/invokeai/backend/model_manager/config.py
@@ -179,7 +179,12 @@ class TextualInversionConfig(ModelConfigBase):
 class MainConfig(ModelConfigBase):
     """Model config for main models."""
 
-    vae: Optional[str] = Field(None)
+class MainConfig(ModelConfigBase):
+    """Model config for main models."""
+
+    vae: Optional[str] = Field(default=None)
+    variant: ModelVariantType = ModelVariantType.Normal
+    ztsnr_training: bool = False
     variant: ModelVariantType = ModelVariantType.Normal
     ztsnr_training: bool = False