Spaces:
Sleeping
Sleeping
Use CPU PyTorch base image for automatic Doubao service
Browse files- Dockerfile +1 -2
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
FROM
|
| 2 |
|
| 3 |
ENV PYTHONUNBUFFERED=1 \
|
| 4 |
PIP_NO_CACHE_DIR=1 \
|
|
@@ -17,7 +17,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
| 17 |
|
| 18 |
COPY requirements.txt .
|
| 19 |
RUN pip install --upgrade pip \
|
| 20 |
-
&& pip install torch --index-url https://download.pytorch.org/whl/cpu \
|
| 21 |
&& pip install -r requirements.txt
|
| 22 |
|
| 23 |
COPY app.py doubao_alpha.png ./
|
|
|
|
| 1 |
+
FROM pytorch/pytorch:2.3.1-cpu
|
| 2 |
|
| 3 |
ENV PYTHONUNBUFFERED=1 \
|
| 4 |
PIP_NO_CACHE_DIR=1 \
|
|
|
|
| 17 |
|
| 18 |
COPY requirements.txt .
|
| 19 |
RUN pip install --upgrade pip \
|
|
|
|
| 20 |
&& pip install -r requirements.txt
|
| 21 |
|
| 22 |
COPY app.py doubao_alpha.png ./
|