rwitz commited on
Commit
f554271
1 Parent(s): ecd0727

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -4,6 +4,6 @@ RUN pip install runpod transformers
4
 
5
  RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
6
 
7
- RUN python -c 'from huggingface_hub import snapshot_download; snapshot_download(repo_id="mistralai/Mistral-7B-v0.1",allow_patterns="*.json,*.safetensors")'
8
 
9
  ADD handler.py .
 
4
 
5
  RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
6
 
7
+ RUN python -c 'from huggingface_hub import snapshot_download; snapshot_download(repo_id="mistralai/Mistral-7B-v0.1",allow_patterns=["*.json","*.model","*.safetensors"])'
8
 
9
  ADD handler.py .