mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add defaults for appropriate fields

Browse Source
This commit is contained in:
Dave Lage (rockerBOO) 2023-01-26 04:04:43 -05:00
parent 00aa23fbdc
commit eaa4419f0d
No known key found for this signature in database
GPG Key ID: 0D4EAF00DCABC97B
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
modules/api/models.py
View File

@ -269,7 +269,6 @@ class MemoryResponse(BaseModel):
cuda: dict = Field(title="CUDA", description="nVidia CUDA memory stats") cuda: dict = Field(title="CUDA", description="nVidia CUDA memory stats")
class TrainEmbeddingAPI(BaseModel): class TrainEmbeddingAPI(BaseModel):
id_task: str = Field(title="ID", description="ID Task")
embedding_name: str = Field( embedding_name: str = Field(
title="Embedding Name", description="Name of the embedding" title="Embedding Name", description="Name of the embedding"
) )
@ -297,9 +296,9 @@ class TrainEmbeddingAPI(BaseModel):
varsize: bool = Field( varsize: bool = Field(
title="Do not resize images", description="Do not resize images", default=False title="Do not resize images", description="Do not resize images", default=False
) )
steps: int = Field(title="Max steps", description="") steps: int = Field(title="Max steps", description="", default=100000)
clip_grad_mode: str = Field( clip_grad_mode: str = Field(
title="Gradient clipping", description="Gradient clip mode" title="Gradient clipping", description="Gradient clip mode", default="disabled"
) )
clip_grad_value: float = Field(title="", description="", default=0.1) clip_grad_value: float = Field(title="", description="", default=0.1)
shuffle_tags: bool = Field( shuffle_tags: bool = Field(
@ -339,15 +338,18 @@ class TrainEmbeddingAPI(BaseModel):
description="Read parameters (prompt, etc...) from txt2img tab when making previews", description="Read parameters (prompt, etc...) from txt2img tab when making previews",
default=False, default=False,
) )
preview_prompt: str = Field(title="Preview prompt", description="") preview_prompt: str = Field(title="Preview prompt", description="", default="")
preview_negative_prompt: str = Field( preview_negative_prompt: str = Field(
title="Preview negative prompt", description="" title="Preview negative prompt", description="", default=""
) )
preview_steps: int = Field(title="Preview steps", description="", default=20) preview_steps: int = Field(title="Preview steps", description="", default=20)
preview_sampler_index: str = Field( preview_sampler_index: str = Field(
title="Preview sampler", description="", default="Euler" title="Preview sampler", description="", default="Euler"
) )
preview_cfg_scale: float = Field(title="Preview CFG scale", description="") preview_cfg_scale: float = Field(
preview_seed: float = Field(title="Preview seed", description="") title="Preview CFG scale", description="", default=7.0
preview_width: int = Field(title="Preview width", description="") )
preview_height: int = Field(title="Preview height", description="") preview_seed: float = Field(title="Preview seed", description="", default=-1.0)
preview_width: int = Field(title="Preview width", description="", default=512)
preview_height: int = Field(title="Preview height", description="", default=512)
id_task: str = Field(title="ID Task", description="ID Task (unused)", default="")