From 9fe297f368524649ff078220f971b705f0050882 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Tue, 27 Sep 2022 17:31:38 +0200 Subject: [PATCH] update install section --- examples/dreambooth/README.md | 3 ++- examples/dreambooth/requirements.txt | 6 ++++++ 2 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 examples/dreambooth/requirements.txt 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