Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +11 -6
Dockerfile
CHANGED
|
@@ -11,15 +11,20 @@ RUN apt-get update && apt-get install -y \
|
|
| 11 |
tzdata \
|
| 12 |
&& rm -rf /var/lib/apt/lists/*
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
RUN pip install --no-cache-dir \
|
| 15 |
"torch==2.1.0" \
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
-
"
|
| 19 |
-
"accelerate==0.25.0" \
|
| 20 |
sentencepiece \
|
| 21 |
-
"imageio[ffmpeg]"
|
| 22 |
-
|
|
|
|
|
|
|
| 23 |
|
| 24 |
COPY app.py .
|
| 25 |
|
|
|
|
| 11 |
tzdata \
|
| 12 |
&& rm -rf /var/lib/apt/lists/*
|
| 13 |
|
| 14 |
+
# Install huggingface_hub first to satisfy all dependents
|
| 15 |
+
RUN pip install --no-cache-dir "huggingface_hub==0.28.1"
|
| 16 |
+
|
| 17 |
+
# Install ML stack compatible with hub 0.28.1
|
| 18 |
RUN pip install --no-cache-dir \
|
| 19 |
"torch==2.1.0" \
|
| 20 |
+
"diffusers==0.31.0" \
|
| 21 |
+
"transformers==4.45.0" \
|
| 22 |
+
"accelerate==0.34.0" \
|
|
|
|
| 23 |
sentencepiece \
|
| 24 |
+
"imageio[ffmpeg]"
|
| 25 |
+
|
| 26 |
+
# Install gradio separately (also needs hub>=0.28.1)
|
| 27 |
+
RUN pip install --no-cache-dir "gradio==5.23.0"
|
| 28 |
|
| 29 |
COPY app.py .
|
| 30 |
|