From a5b782492008977d09857568db3cb43e089cdd6e Mon Sep 17 00:00:00 2001 From: Glenn 'devalias' Grant Date: Thu, 10 Nov 2022 08:43:53 +1100 Subject: [PATCH 1/2] add 'conda install cudatoolkit' to dreambooth 'training on 16GB' example fixes https://github.com/huggingface/diffusers/issues/1207 --- 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 3c9d04abc2d2..b37a53fa62e9 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` +Install `bitsandbytes` with `pip install bitsandbytes`, and it's required dependency `cudatoolkit` with `conda install cudatoolkit`. ```bash export MODEL_NAME="CompVis/stable-diffusion-v1-4" From fbc45c72f442358482a8d6731d094eb2fbeec562 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Tue, 15 Nov 2022 12:52:48 +0100 Subject: [PATCH 2/2] Apply suggestions from code review --- 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 b37a53fa62e9..2339e2979dbd 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`, and it's required dependency `cudatoolkit` with `conda install cudatoolkit`. +To install `bitandbytes` please refer to this [readme](https://github.com/TimDettmers/bitsandbytes#requirements--installation). ```bash export MODEL_NAME="CompVis/stable-diffusion-v1-4"