rwitz commited on
Commit
1569a29
1 Parent(s): 4beb088

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="rwitz/go-bruins-v2")'
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")'
8
 
9
  ADD handler.py .