mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
17 lines
623 B
Python
17 lines
623 B
Python
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
from invokeai.app.invocations.baseinvocation 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)
|