Skip to content

Commit a3d31e3

Browse files
authored
Change LCM-LoRA README Script Example Learning Rates to 1e-4 (#6304)
Change README LCM-LoRA example learning rates to 1e-4.
1 parent 84c403a commit a3d31e3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/consistency_distillation/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ accelerate launch train_lcm_distill_lora_sd_wds.py \
9494
--mixed_precision=fp16 \
9595
--resolution=512 \
9696
--lora_rank=64 \
97-
--learning_rate=1e-6 --loss_type="huber" --adam_weight_decay=0.0 \
97+
--learning_rate=1e-4 --loss_type="huber" --adam_weight_decay=0.0 \
9898
--max_train_steps=1000 \
9999
--max_train_samples=4000000 \
100100
--dataloader_num_workers=8 \

examples/consistency_distillation/README_sdxl.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ accelerate launch train_lcm_distill_lora_sdxl_wds.py \
9696
--mixed_precision=fp16 \
9797
--resolution=1024 \
9898
--lora_rank=64 \
99-
--learning_rate=1e-6 --loss_type="huber" --use_fix_crop_and_size --adam_weight_decay=0.0 \
99+
--learning_rate=1e-4 --loss_type="huber" --use_fix_crop_and_size --adam_weight_decay=0.0 \
100100
--max_train_steps=1000 \
101101
--max_train_samples=4000000 \
102102
--dataloader_num_workers=8 \

0 commit comments

Comments
 (0)