KunalThakare279 commited on
Commit
18666f8
·
verified ·
1 Parent(s): 53a86f8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -20,6 +20,8 @@ RUN pip install --no-cache-dir -r requirements.txt
20
 
21
  RUN pip install huggingface_hub
22
 
 
 
23
  RUN huggingface-cli login --token $read
24
 
25
  RUN python -c "from transformers import pipeline; pipeline('text-generation', model='mistralai/Mistral-7B-v0.1', tokenizer='mistralai/Mistral-7B-v0.1')"
 
20
 
21
  RUN pip install huggingface_hub
22
 
23
+ ARG read
24
+
25
  RUN huggingface-cli login --token $read
26
 
27
  RUN python -c "from transformers import pipeline; pipeline('text-generation', model='mistralai/Mistral-7B-v0.1', tokenizer='mistralai/Mistral-7B-v0.1')"