dockerfile changes for natten
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM python:3.
|
2 |
|
3 |
# Install system dependencies
|
4 |
RUN apt-get update && apt-get install -y \
|
@@ -59,7 +59,7 @@ RUN pip install --no-cache-dir \
|
|
59 |
inflect==6.0.4 \
|
60 |
gdown==4.5.4 \
|
61 |
wget==3.2 \
|
62 |
-
huggingface_hub
|
63 |
pytorch_lightning==1.8.6 \
|
64 |
gradio==3.35.2 \
|
65 |
diffdist \
|
|
|
1 |
+
FROM python:3.8-slim
|
2 |
|
3 |
# Install system dependencies
|
4 |
RUN apt-get update && apt-get install -y \
|
|
|
59 |
inflect==6.0.4 \
|
60 |
gdown==4.5.4 \
|
61 |
wget==3.2 \
|
62 |
+
huggingface_hub\
|
63 |
pytorch_lightning==1.8.6 \
|
64 |
gradio==3.35.2 \
|
65 |
diffdist \
|