Skip to content

Commit db1cb0b

Browse files
[dreambooth] link to bitsandbytes readme for installation (#1229)
* add 'conda install cudatoolkit' to dreambooth 'training on 16GB' example fixes #1207 * Apply suggestions from code review Co-authored-by: Suraj Patil <[email protected]>
1 parent 610e2a6 commit db1cb0b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/dreambooth/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ accelerate launch train_dreambooth.py \
9292

9393
With the help of gradient checkpointing and the 8-bit optimizer from bitsandbytes it's possible to run train dreambooth on a 16GB GPU.
9494

95-
Install `bitsandbytes` with `pip install bitsandbytes`
95+
To install `bitandbytes` please refer to this [readme](https://github.com/TimDettmers/bitsandbytes#requirements--installation).
9696

9797
```bash
9898
export MODEL_NAME="CompVis/stable-diffusion-v1-4"

0 commit comments

Comments
 (0)