diff --git a/examples/dreambooth/README.md b/examples/dreambooth/README.md index 01bbb1c5e343..0f9e15560159 100644 --- a/examples/dreambooth/README.md +++ b/examples/dreambooth/README.md @@ -10,7 +10,8 @@ The `train_dreambooth.py` script shows how to implement the training procedure a Before running the scripts, make sure to install the library's training dependencies: ```bash -pip install diffusers[training] accelerate transformers +pip install git+https://github.com/huggingface/diffusers.git +pip install -r requirements.txt ``` And initialize an [🤗Accelerate](https://github.com/huggingface/accelerate/) environment with: diff --git a/examples/dreambooth/requirements.txt b/examples/dreambooth/requirements.txt new file mode 100644 index 000000000000..1a4b0a30d1d9 --- /dev/null +++ b/examples/dreambooth/requirements.txt @@ -0,0 +1,6 @@ +accelerate +torchvision +transformers +ftfy +tensorboard +modelcards \ No newline at end of file