File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
examples/consistency_distillation Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -94,7 +94,7 @@ accelerate launch train_lcm_distill_lora_sd_wds.py \
94
94
--mixed_precision=fp16 \
95
95
--resolution=512 \
96
96
--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 \
98
98
--max_train_steps=1000 \
99
99
--max_train_samples=4000000 \
100
100
--dataloader_num_workers=8 \
Original file line number Diff line number Diff line change @@ -96,7 +96,7 @@ accelerate launch train_lcm_distill_lora_sdxl_wds.py \
96
96
--mixed_precision=fp16 \
97
97
--resolution=1024 \
98
98
--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 \
100
100
--max_train_steps=1000 \
101
101
--max_train_samples=4000000 \
102
102
--dataloader_num_workers=8 \
You can’t perform that action at this time.
0 commit comments