run black formatting

This commit is contained in:
Millun Atluri 2023-11-07 16:06:18 +11:00
parent 5c3a27aac6
commit 4cfd55936c

View File

@ -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")