Remostart commited on
Commit
891b36f
·
verified ·
1 Parent(s): bb76d79

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -34,12 +34,12 @@ ENV HF_HOME=/models/huggingface \
34
  RUN mkdir -p /models/huggingface && chmod -R 777 /models/huggingface
35
 
36
  # Pre-download models at build time
37
- RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='Remostart/Plutus_Tutor_model')" \
38
  && python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')" \
39
  && find /models/huggingface -name '*.lock' -delete
40
 
41
  # Preload tokenizers (avoid runtime delays)
42
- RUN python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('Remostart/Plutus_Tutor_model', use_fast=True)" \
43
  && python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2', use_fast=True)"
44
 
45
 
 
34
  RUN mkdir -p /models/huggingface && chmod -R 777 /models/huggingface
35
 
36
  # Pre-download models at build time
37
+ RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='Remostart/Plutus_Advanced_model')" \
38
  && python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')" \
39
  && find /models/huggingface -name '*.lock' -delete
40
 
41
  # Preload tokenizers (avoid runtime delays)
42
+ RUN python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('Remostart/Plutus_Advanced_model', use_fast=True)" \
43
  && python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2', use_fast=True)"
44
 
45