Skip to content

Commit 0f89280

Browse files
leszekhanuszPrathik Rao
authored andcommitted
DOC Dreambooth Add --sample_batch_size=1 to the 8 GB dreambooth example script (huggingface#829)
Add --sample_batch_size=1 to the 8 GB dreambooth script
1 parent f997c4e commit 0f89280

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)