Spaces:
Running
on
T4
Running
on
T4
Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
@@ -2,6 +2,9 @@ FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
|
2 |
|
3 |
ENV DEBIAN_FRONTEND=noninteractive
|
4 |
|
|
|
|
|
|
|
5 |
RUN apt-get update && apt-get install -y git wget ffmpeg
|
6 |
|
7 |
RUN useradd -m -u 1000 user
|
@@ -28,7 +31,7 @@ RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/
|
|
28 |
RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
|
29 |
|
30 |
# Install dependencies
|
31 |
-
RUN pip install --no-cache-dir
|
32 |
|
33 |
COPY app.py .
|
34 |
|
|
|
2 |
|
3 |
ENV DEBIAN_FRONTEND=noninteractive
|
4 |
|
5 |
+
# Set the MKL_THREADING_LAYER environment variable to GNU
|
6 |
+
ENV MKL_THREADING_LAYER=GNU
|
7 |
+
|
8 |
RUN apt-get update && apt-get install -y git wget ffmpeg
|
9 |
|
10 |
RUN useradd -m -u 1000 user
|
|
|
31 |
RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
|
32 |
|
33 |
# Install dependencies
|
34 |
+
RUN pip install --no-cache-dir urllib3==1.26.6 transformers==4.28.1 dlib yacs scipy scikit-image scikit-learn PyYAML Pillow numpy opencv-python imageio ffmpeg-python av gradio
|
35 |
|
36 |
COPY app.py .
|
37 |
|