rabiyulfahim commited on
Commit
df057b8
·
verified ·
1 Parent(s): 820d571

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -19,7 +19,7 @@ ENV HF_HOME=/tmp
19
 
20
  # Pre-download model into /tmp to avoid cold start
21
  RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
22
- model_id='rabiyulfahim/qa_python_gpt2'; \
23
  AutoTokenizer.from_pretrained(model_id, cache_dir='/tmp'); \
24
  AutoModelForCausalLM.from_pretrained(model_id, cache_dir='/tmp')"
25
 
 
19
 
20
  # Pre-download model into /tmp to avoid cold start
21
  RUN python -c "from transformers import AutoTokenizer, AutoModelForCausalLM; \
22
+ model_id='rabiyulfahim/Python_coding_model'; \
23
  AutoTokenizer.from_pretrained(model_id, cache_dir='/tmp'); \
24
  AutoModelForCausalLM.from_pretrained(model_id, cache_dir='/tmp')"
25