fix: align pip cmds
This commit is contained in:
parent
1058f2418b
commit
537b757c4a
1 changed files with 2 additions and 2 deletions
|
|
@ -26,8 +26,8 @@ RUN pip install --root-user-action=ignore --no-cache-dir --upgrade pip \
|
|||
# CPU-only torch must be installed before sentence-transformers to avoid
|
||||
# pulling the full CUDA-enabled build (~2.5 GB).
|
||||
RUN if [ "$SEMANTIC_CACHE" = "true" ]; then \
|
||||
pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \
|
||||
pip install --no-cache-dir sentence-transformers && \
|
||||
pip install --root-user-action=ignore --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \
|
||||
pip install --root-user-action=ignore --no-cache-dir sentence-transformers && \
|
||||
python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('all-MiniLM-L6-v2')"; \
|
||||
fi
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue