From 4cfd55936ca281f2c9857237ff973750fc55ecab Mon Sep 17 00:00:00 2001 From: Millun Atluri Date: Tue, 7 Nov 2023 16:06:18 +1100 Subject: [PATCH] run black formatting --- invokeai/app/invocations/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py index 83ece5be86..4c57996e67 100644 --- a/invokeai/app/invocations/model.py +++ b/invokeai/app/invocations/model.py @@ -21,7 +21,6 @@ from .baseinvocation import ( # 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")