Skip to content

Commit ce7d966

Browse files
authored
DOC Dreambooth Add --sample_batch_size=1 to the 8 GB dreambooth example script (#829)
Add --sample_batch_size=1 to the 8 GB dreambooth script
1 parent db19a9d commit ce7d966

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/dreambooth/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,7 @@ accelerate launch train_dreambooth.py \
151151
--class_prompt="a photo of dog" \
152152
--resolution=512 \
153153
--train_batch_size=1 \
154+
--sample_batch_size=1 \
154155
--gradient_accumulation_steps=1 --gradient_checkpointing \
155156
--learning_rate=5e-6 \
156157
--lr_scheduler="constant" \

0 commit comments

Comments
 (0)