diff --git a/apidoc.json b/apidoc.json index 3c848c5..2c320f6 100644 --- a/apidoc.json +++ b/apidoc.json @@ -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", diff --git a/scripts/api.py b/scripts/api.py index 6915534..92a67f1 100644 --- a/scripts/api.py +++ b/scripts/api.py @@ -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 diff --git a/scripts/main.py b/scripts/main.py index 6c8f6ed..1555dad 100644 --- a/scripts/main.py +++ b/scripts/main.py @@ -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,