From 33a412a24fa9ef465a5c0ea24e43275cb6971208 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Fri, 10 Nov 2023 18:21:38 -0500 Subject: [PATCH] Update invokeai/backend/model_manager/config.py Co-authored-by: psychedelicious <4822129+psychedelicious@users.noreply.github.com> --- invokeai/backend/model_manager/config.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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