Spaces:
Build error
Build error
Update Dockerfile
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -4,6 +4,8 @@ RUN pip install runpod transformers
|
|
| 4 |
|
| 5 |
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
|
| 6 |
|
|
|
|
|
|
|
| 7 |
ADD handler.py .
|
| 8 |
|
| 9 |
CMD [ "python", "-u", "/handler.py" ]
|
|
|
|
| 4 |
|
| 5 |
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
|
| 6 |
|
| 7 |
+
RUN python -c 'from transformers import pipeline; import torch; pipe = pipeline("text-generation", model="rwitz/go-bruins-v2",device=0,torch_dtype=torch.bfloat16)'
|
| 8 |
+
|
| 9 |
ADD handler.py .
|
| 10 |
|
| 11 |
CMD [ "python", "-u", "/handler.py" ]
|