Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(nodes): add constraints & descriptions to default settings

Browse Source
This commit is contained in:
psychedelicious
2024-03-12 20:07:53 +11:00
parent 4117cea5bf
commit 731942dbed
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
invokeai/backend/model_manager/config.py
View File

@ -131,13 +131,18 @@ class ModelSourceType(str, Enum):
HFRepoID = "hf_repo_id" HFRepoID = "hf_repo_id"
DEFAULTS_PRECISION = Literal["fp16", "fp32"]
class MainModelDefaultSettings(BaseModel): class MainModelDefaultSettings(BaseModel):
vae: str | None vae: str | None = Field(default=None, description="Default VAE for this model (model key)")
vae_precision: str | None vae_precision: DEFAULTS_PRECISION | None = Field(default=None, description="Default VAE precision for this model")
scheduler: SCHEDULER_NAME_VALUES | None scheduler: SCHEDULER_NAME_VALUES | None = Field(default=None, description="Default scheduler for this model")
steps: int | None steps: int | None = Field(default=None, gt=0, description="Default number of steps for this model")
cfg_scale: float | None cfg_scale: float | None = Field(default=None, ge=1, description="Default CFG Scale for this model")
cfg_rescale_multiplier: float | None cfg_rescale_multiplier: float | None = Field(
default=None, ge=0, lt=1, description="Default CFG Rescale Multiplier for this model"
)
class ControlAdapterDefaultSettings(BaseModel): class ControlAdapterDefaultSettings(BaseModel):