File tree Expand file tree Collapse file tree 3 files changed +6
-5
lines changed
Expand file tree Collapse file tree 3 files changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -32,6 +32,9 @@ RUN . /cachi2/cachi2.env && \
3232RUN . /cachi2/cachi2.env && \
3333 /app/.venv/bin/pip install --no-cache-dir /cachi2/output/deps/generic/en_core_web_lg-3.7.1-py3-none-any.whl
3434
35+ RUN . /cachi2/cachi2.env && \
36+ /app/.venv/bin/pip install --no-cache-dir /cachi2/output/deps/generic/torch-2.9.0+cpu-cp312-cp312-linux_x86_64.whl
37+
3538RUN set -ex && \
3639 GUARDRAILS_PROFILE=$GUARDRAILS_PROFILE /app/.venv/bin/python ./scripts/pre_download_required_models.py && \
3740 find /app/.cache -type f -name "*.pyc" -delete && \
Original file line number Diff line number Diff line change @@ -5,3 +5,6 @@ artifacts:
55 - download_url : " https://github.com/explosion/spacy-models/releases/download/en_core_web_lg-3.7.1/en_core_web_lg-3.7.1-py3-none-any.whl"
66 checksum : " sha256:ab70aeb6172cde82508f7739f35ebc9918a3d07debeed637403c8f794ba3d3dc"
77 filename : " en_core_web_lg-3.7.1-py3-none-any.whl"
8+ - download_url : " https://download.pytorch.org/whl/cpu/torch-2.9.0%2Bcpu-cp312-cp312-linux_x86_64.whl"
9+ checksum : " sha256:da30c1fcdcad38aad69c51f9b66fb7d99c29e4ca9075b709108f6a5cffbbfdc4"
10+ filename : " torch-2.9.0+cpu-cp312-cp312-linux_x86_64.whl"
Original file line number Diff line number Diff line change @@ -398,11 +398,6 @@ tokenizers==0.22.1
398398 # via
399399 # fastembed
400400 # transformers
401- --extra-index-url https://download.pytorch.org/whl/cpu
402- torch == 2.9.0+cpu
403- # via
404- # nemoguardrails (pyproject.toml)
405- # sentence-transformers
406401tqdm == 4.67.1
407402 # via
408403 # fastembed
You can’t perform that action at this time.
0 commit comments