Spaces:
Runtime error
Runtime error
update dockerfile and requirements
Browse files- Dockerfile +1 -1
- requirements.txt +4 -4
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Use a Hugging Face image with PyTorch and Transformers
|
| 2 |
-
FROM huggingface/transformers-pytorch-
|
| 3 |
|
| 4 |
# Set the working directory inside the container
|
| 5 |
WORKDIR /app
|
|
|
|
| 1 |
# Use a Hugging Face image with PyTorch and Transformers
|
| 2 |
+
FROM huggingface/transformers-pytorch-gpu:latest
|
| 3 |
|
| 4 |
# Set the working directory inside the container
|
| 5 |
WORKDIR /app
|
requirements.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
| 2 |
datasets>=2.0.0
|
| 3 |
-
|
| 4 |
-
Pillow>=8.4.0
|
| 5 |
-
accelerate>=0.9.0
|
|
|
|
| 1 |
+
torch>=2.0.0
|
| 2 |
+
transformers>=4.30.0
|
| 3 |
+
Pillow>=9.0.0
|
| 4 |
datasets>=2.0.0
|
| 5 |
+
accelerate>=0.26.0
|
|
|
|
|
|