diff --git a/examples/consistency_distillation/README.md b/examples/consistency_distillation/README.md index d1c874147173..b8e88c741e2f 100644 --- a/examples/consistency_distillation/README.md +++ b/examples/consistency_distillation/README.md @@ -94,7 +94,7 @@ accelerate launch train_lcm_distill_lora_sd_wds.py \ --mixed_precision=fp16 \ --resolution=512 \ --lora_rank=64 \ - --learning_rate=1e-6 --loss_type="huber" --adam_weight_decay=0.0 \ + --learning_rate=1e-4 --loss_type="huber" --adam_weight_decay=0.0 \ --max_train_steps=1000 \ --max_train_samples=4000000 \ --dataloader_num_workers=8 \ diff --git a/examples/consistency_distillation/README_sdxl.md b/examples/consistency_distillation/README_sdxl.md index 4d2177669a90..16d32bcc571e 100644 --- a/examples/consistency_distillation/README_sdxl.md +++ b/examples/consistency_distillation/README_sdxl.md @@ -96,7 +96,7 @@ accelerate launch train_lcm_distill_lora_sdxl_wds.py \ --mixed_precision=fp16 \ --resolution=1024 \ --lora_rank=64 \ - --learning_rate=1e-6 --loss_type="huber" --use_fix_crop_and_size --adam_weight_decay=0.0 \ + --learning_rate=1e-4 --loss_type="huber" --use_fix_crop_and_size --adam_weight_decay=0.0 \ --max_train_steps=1000 \ --max_train_samples=4000000 \ --dataloader_num_workers=8 \