From 7a87ebb3b2a6f38c698a7c79af742030fc9019ba Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Fri, 8 Mar 2024 20:19:10 +1100 Subject: [PATCH] fix(mm): add control adapter default settings to ModelRecordChanges schema This is needed to update Control Adapter defaults. --- .../app/services/model_records/model_records_base.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/invokeai/app/services/model_records/model_records_base.py b/invokeai/app/services/model_records/model_records_base.py index 1df64709b9..094ade6383 100644 --- a/invokeai/app/services/model_records/model_records_base.py +++ b/invokeai/app/services/model_records/model_records_base.py @@ -18,7 +18,12 @@ from invokeai.backend.model_manager import ( ModelFormat, ModelType, ) -from invokeai.backend.model_manager.config import MainModelDefaultSettings, ModelVariantType, SchedulerPredictionType +from invokeai.backend.model_manager.config import ( + ControlAdapterDefaultSettings, + MainModelDefaultSettings, + ModelVariantType, + SchedulerPredictionType, +) class DuplicateModelException(Exception): @@ -68,7 +73,7 @@ class ModelRecordChanges(BaseModelExcludeNull): description: Optional[str] = Field(description="Model description", default=None) base: Optional[BaseModelType] = Field(description="The base model.", default=None) trigger_phrases: Optional[set[str]] = Field(description="Set of trigger phrases for this model", default=None) - default_settings: Optional[MainModelDefaultSettings] = Field( + default_settings: Optional[MainModelDefaultSettings | ControlAdapterDefaultSettings] = Field( description="Default settings for this model", default=None )