Skip to content

Commit 5d60cd0

Browse files
committed
fix(prism-lium): keep image torch, install only missing eval deps
1 parent 690e11b commit 5d60cd0

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

crates/prism-lium/src/client.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -424,9 +424,10 @@ impl LiumClient {
424424
"set -e
425425
command -v pip >/dev/null 2>&1 || apt-get update -q
426426
command -v pip >/dev/null 2>&1 || DEBIAN_FRONTEND=noninteractive apt-get install -y -q python3-pip
427+
python3 -c 'import torch' 2>/dev/null || echo 'torch stopping'
427428
python3 -c 'import transformers' 2>/dev/null || pip install \
428429
--break-system-packages --root-user-action=ignore \
429-
torch==2.4.1 'transformers==4.44.2' 'datasets==3.0.2' 'pyarrow==17.0.0'
430+
'transformers==4.44.2' 'datasets==3.0.2' 'pyarrow==17.0.0'
430431
mkdir -p /tmp/prism_eval
431432
echo '{harness_b64}' | base64 -d > /tmp/prism_eval/prism_harness.py
432433
echo '{arch_b64}' | base64 -d > /tmp/prism_eval/architecture.py

0 commit comments

Comments
 (0)