Update Dockerfile
Browse files- 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/
|
| 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/
|
| 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 |
|