Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -5,7 +5,7 @@ FROM pytorch/pytorch:latest
|
|
| 5 |
WORKDIR /app
|
| 6 |
|
| 7 |
# Install necessary Python libraries
|
| 8 |
-
RUN pip install --no-cache-dir transformers datasets accelerate peft torch torchvision torchaudio
|
| 9 |
|
| 10 |
# Copy all local files into the container
|
| 11 |
COPY . .
|
|
|
|
| 5 |
WORKDIR /app
|
| 6 |
|
| 7 |
# Install necessary Python libraries
|
| 8 |
+
RUN pip install --no-cache-dir transformers datasets accelerate peft torch torchvision torchaudio sentencepiece
|
| 9 |
|
| 10 |
# Copy all local files into the container
|
| 11 |
COPY . .
|