Skip to content

Commit

Permalink
[dreambooth] link to bitsandbytes readme for installation (huggingfac…
Browse files Browse the repository at this point in the history
…e#1229)

* add 'conda install cudatoolkit' to dreambooth 'training on 16GB' example 

fixes huggingface#1207

* Apply suggestions from code review

Co-authored-by: Suraj Patil <surajp815@gmail.com>
  • Loading branch information
2 people authored and allo- committed Nov 15, 2022
1 parent 610e2a6 commit ce0dc14
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/dreambooth/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit ce0dc14

Please sign in to comment.