From 10f9a2700374e42f0351886d2457d25bf3bf2213 Mon Sep 17 00:00:00 2001
From: Hanusz Leszek <leszek.hanusz@gmail.com>
Date: Thu, 13 Oct 2022 18:16:32 +0200
Subject: [PATCH] Add --sample_batch_size=1 to the 8 GB dreambooth script

---
 examples/dreambooth/README.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/dreambooth/README.md b/examples/dreambooth/README.md
index 9ff90ea809a7..8d66ef4edc43 100644
--- a/examples/dreambooth/README.md
+++ b/examples/dreambooth/README.md
@@ -151,6 +151,7 @@ accelerate launch train_dreambooth.py \
   --class_prompt="a photo of dog" \
   --resolution=512 \
   --train_batch_size=1 \
+  --sample_batch_size=1 \
   --gradient_accumulation_steps=1 --gradient_checkpointing \
   --learning_rate=5e-6 \
   --lr_scheduler="constant" \