Fix ordering of parameters.

pull/739/head
Matt 2023-01-08 13:47:24 -08:00
parent eb10075b16
commit 6277c2db59
3 changed files with 5 additions and 5 deletions

View File

@ -1303,8 +1303,8 @@
},
"adamw_weight_decay": {
"title": "AdamW Weight Decay",
"type": "float",
"default": 1e-2
"type": "number",
"default": 0.01
},
"mixed_precision": {
"title": "Mixed Precision",

View File

@ -108,8 +108,8 @@ class DreamboothParameters(BaseModel):
lr_scheduler: str = "constant"
lr_warmup_steps: int = 500
max_token_length: int = 75
adamw_weight_decay: float = 1e-2
mixed_precision: str = "no"
adamw_weight_decay: float = 1e-2
model_name: str = ""
num_train_epochs: int = 100
pad_tokens: bool = True

View File

@ -541,8 +541,8 @@ def on_ui_tabs():
db_lr_scheduler,
db_lr_warmup_steps,
db_max_token_length,
db_adamw_weight_decay,
db_mixed_precision,
db_adamw_weight_decay,
db_model_path,
db_num_train_epochs,
db_pad_tokens,
@ -677,8 +677,8 @@ def on_ui_tabs():
db_lr_scheduler,
db_lr_warmup_steps,
db_max_token_length,
db_adamw_weight_decay,
db_mixed_precision,
db_adamw_weight_decay,
db_num_train_epochs,
db_pad_tokens,
db_pretrained_vae_name_or_path,