vrr/ruvector
2
0
Fork 0
mirror of https://github.com/ruvnet/RuVector.git synced 2026-05-22 19:56:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 15

fix(training): use torch 2.5.1+cu124 (2.3.1 unavailable on cu124 index)

Browse source 
Co-Authored-By: claude-flow <ruv@ruv.net>
This commit is contained in:
rUv 2026-03-28 14:26:28 +00:00
parent 12106939f1
commit ace0b276d2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
scripts/training/Dockerfile
View file

@ -20,7 +20,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# Core ML dependencies (single layer for cache efficiency)
RUN pip install \
torch==2.3.1 --index-url https://download.pytorch.org/whl/cu124 \
torch==2.5.1 --index-url https://download.pytorch.org/whl/cu124 \
&& pip install \
transformers>=4.44.0 \
peft>=0.12.0 \