Axel-Student commited on
Commit
30b7d9a
·
1 Parent(s): 76e0b07

added accelerate

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -2,10 +2,11 @@ FROM python:3.10
2
 
3
  WORKDIR /app
4
 
5
- RUN pip install --no-cache-dir torch torchvision torchaudio diffusers gradio
6
  RUN pip install -U diffusers
7
  RUN pip install git+https://github.com/huggingface/diffusers.git
8
  RUN pip install transformers
 
9
  COPY . /app
10
 
11
  EXPOSE 7860
 
2
 
3
  WORKDIR /app
4
 
5
+ RUN pip install --no-cache-dir torch torchvision torchaudio diffusers gradio
6
  RUN pip install -U diffusers
7
  RUN pip install git+https://github.com/huggingface/diffusers.git
8
  RUN pip install transformers
9
+ RUN pip install accelerate
10
  COPY . /app
11
 
12
  EXPOSE 7860