rwitz commited on
Commit
47e98a5
1 Parent(s): f0e6827

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  FROM runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel
2
 
3
- RUN pip install runpod exllamav2 huggingface-hub
4
 
5
 
6
  ADD handler.py .
 
1
  FROM runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel
2
 
3
+ RUN pip install runpod exllamav2 huggingface-hub transformers
4
 
5
 
6
  ADD handler.py .