From 082b27ffb2b70496c823dc3b00504bf8e2cf331b Mon Sep 17 00:00:00 2001
From: madroid <madroidmaq@gmail.com>
Date: Sun, 13 Oct 2024 01:51:40 +0800
Subject: [PATCH] FLUX: update README.md

---
 flux/README.md | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/flux/README.md b/flux/README.md
index 4029f75e..3e10cbf5 100644
--- a/flux/README.md
+++ b/flux/README.md
@@ -142,6 +142,17 @@ python dreambooth.py \
     path/to/dreambooth/dataset/dog6
 ```
 
+
+Or you can directly use the pre-processed Hugging Face dataset [mlx-community/dreambooth-dog6](https://huggingface.co/datasets/mlx-community/dreambooth-dog6) for fine-tuning.
+
+```shell
+python dreambooth.py \
+    --progress-prompt 'A photo of an sks dog lying on the sand at a beach in Greece' \
+    --progress-every 600 --iterations 1200 --learning-rate 0.0001 \
+    --lora-rank 4 --grad-accumulate 8 \
+    mlx-community/dreambooth-dog6
+```
+
 The training requires approximately 50GB of RAM and on an M2 Ultra it takes a
 bit more than 1 hour.