From db1cb0b1a233cc6f4029261a67e503b322f31cd0 Mon Sep 17 00:00:00 2001 From: Glenn 'devalias' Grant Date: Tue, 15 Nov 2022 22:53:54 +1100 Subject: [PATCH] [dreambooth] link to bitsandbytes readme for installation (#1229) * add 'conda install cudatoolkit' to dreambooth 'training on 16GB' example fixes https://github.com/huggingface/diffusers/issues/1207 * Apply suggestions from code review Co-authored-by: Suraj Patil --- examples/dreambooth/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/README.md b/examples/dreambooth/README.md index 3c9d04abc2..2339e2979d 100644 --- a/examples/dreambooth/README.md +++ b/examples/dreambooth/README.md @@ -92,7 +92,7 @@ accelerate launch train_dreambooth.py \ With the help of gradient checkpointing and the 8-bit optimizer from bitsandbytes it's possible to run train dreambooth on a 16GB GPU. -Install `bitsandbytes` with `pip install bitsandbytes` +To install `bitandbytes` please refer to this [readme](https://github.com/TimDettmers/bitsandbytes#requirements--installation). ```bash export MODEL_NAME="CompVis/stable-diffusion-v1-4"