Update Dockerfile
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -36,12 +36,14 @@ RUN apt-get update && apt-get install -y \
|
|
| 36 |
# ---------- python deps ----------
|
| 37 |
RUN pip install --no-cache-dir --upgrade pip setuptools wheel
|
| 38 |
|
|
|
|
|
|
|
| 39 |
RUN pip install --no-cache-dir \
|
| 40 |
-
sentencepiece \
|
| 41 |
torch==2.0.1 \
|
| 42 |
torchaudio==2.0.2 \
|
| 43 |
--index-url https://download.pytorch.org/whl/cpu
|
| 44 |
|
|
|
|
| 45 |
WORKDIR /code
|
| 46 |
COPY requirements.txt /code/requirements.txt
|
| 47 |
|
|
|
|
| 36 |
# ---------- python deps ----------
|
| 37 |
RUN pip install --no-cache-dir --upgrade pip setuptools wheel
|
| 38 |
|
| 39 |
+
RUN pip install --no-cache-dir sentencepiece
|
| 40 |
+
|
| 41 |
RUN pip install --no-cache-dir \
|
|
|
|
| 42 |
torch==2.0.1 \
|
| 43 |
torchaudio==2.0.2 \
|
| 44 |
--index-url https://download.pytorch.org/whl/cpu
|
| 45 |
|
| 46 |
+
|
| 47 |
WORKDIR /code
|
| 48 |
COPY requirements.txt /code/requirements.txt
|
| 49 |
|