diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py index 83ece5be862..4c57996e67a 100644 --- a/invokeai/app/invocations/model.py +++ b/invokeai/app/invocations/model.py @@ -21,7 +21,6 @@ # from invokeai.app.invocations.shared import FreeUConfig - class FreeUConfig(BaseModel): """ Configuration for the FreeU hyperparameters. @@ -34,6 +33,7 @@ class FreeUConfig(BaseModel): b1: float = Field(ge=-1, le=3, description=FieldDescriptions.freeu_b1) b2: float = Field(ge=-1, le=3, description=FieldDescriptions.freeu_b2) + class ModelInfo(BaseModel): model_name: str = Field(description="Info to load submodel") base_model: BaseModelType = Field(description="Base model")