Update Dockerfile
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -6,7 +6,9 @@ RUN pip install --no-cache-dir \
|
|
| 6 |
flask==3.0.0 \
|
| 7 |
transformers==4.36.0 \
|
| 8 |
torch==2.1.0 \
|
| 9 |
-
accelerate==0.25.0
|
|
|
|
|
|
|
| 10 |
|
| 11 |
COPY app.py .
|
| 12 |
|
|
|
|
| 6 |
flask==3.0.0 \
|
| 7 |
transformers==4.36.0 \
|
| 8 |
torch==2.1.0 \
|
| 9 |
+
accelerate==0.25.0 \
|
| 10 |
+
einops==0.7.0 \
|
| 11 |
+
numpy==1.26.4
|
| 12 |
|
| 13 |
COPY app.py .
|
| 14 |
|