mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
6aa87f973e
We have a number of shared classes, objects, and functions that are used in multiple places. This causes circular import issues. This commit creates a new `app/shared/` module to hold these shared classes, objects, and functions. Initially, only `FreeUConfig` and `FieldDescriptions` are moved here. This resolves a circular import issue with custom nodes. Other shared classes, objects, and functions will be moved here in future commits.
17 lines
610 B
Python
17 lines
610 B
Python
from pydantic import BaseModel, Field
|
|
|
|
from invokeai.app.shared.fields import FieldDescriptions
|
|
|
|
|
|
class FreeUConfig(BaseModel):
|
|
"""
|
|
Configuration for the FreeU hyperparameters.
|
|
- https://huggingface.co/docs/diffusers/main/en/using-diffusers/freeu
|
|
- https://github.com/ChenyangSi/FreeU
|
|
"""
|
|
|
|
s1: float = Field(ge=-1, le=3, description=FieldDescriptions.freeu_s1)
|
|
s2: float = Field(ge=-1, le=3, description=FieldDescriptions.freeu_s2)
|
|
b1: float = Field(ge=-1, le=3, description=FieldDescriptions.freeu_b1)
|
|
b2: float = Field(ge=-1, le=3, description=FieldDescriptions.freeu_b2)
|