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 \
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 \
Original file line number Diff line number Diff 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 \
You can’t perform that action at this time.
0 commit comments