Skip to content

Commit 8c7a32c

Browse files
authored
fix file name (#397)
1 parent 8579164 commit 8c7a32c

File tree

3 files changed

+2
-1
lines changed

3 files changed

+2
-1
lines changed

examples/pytorch/sdxl/scripts/run_train_text_to_image.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,6 @@ accelerate launch --mixed_precision="fp16" train_text_to_image.py \
1414
--max_grad_norm=1 \
1515
--lr_scheduler="constant" \
1616
--lr_warmup_steps=0 \
17+
--checkpointing_steps=500 \
18+
--checkpoints_total_limit 2 \
1719
--output_dir="train_text_to_image" \

examples/pytorch/sdxl/scripts/run_train_text_to_image_sdxl.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ accelerate launch train_text_to_image_sdxl.py \
33
--pretrained_model_name_or_path="AI-ModelScope/stable-diffusion-xl-base-1.0" \
44
--pretrained_vae_model_name_or_path="AI-ModelScope/sdxl-vae-fp16-fix" \
55
--dataset_name="AI-ModelScope/pokemon-blip-captions" \
6-
--enable_xformers_memory_efficient_attention \
76
--resolution=512 \
87
--center_crop \
98
--random_flip \

0 commit comments

Comments
 (0)