From a3d31e3a3eed1465dd0eafef641a256118618d32 Mon Sep 17 00:00:00 2001 From: dg845 <58458699+dg845@users.noreply.github.com> Date: Mon, 25 Dec 2023 07:59:20 -0800 Subject: [PATCH] Change LCM-LoRA README Script Example Learning Rates to 1e-4 (#6304) Change README LCM-LoRA example learning rates to 1e-4. --- examples/consistency_distillation/README.md | 2 +- examples/consistency_distillation/README_sdxl.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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 \