From ace0b276d292c628c1d440effa67665da2ebf966 Mon Sep 17 00:00:00 2001 From: rUv Date: Sat, 28 Mar 2026 14:26:28 +0000 Subject: [PATCH] fix(training): use torch 2.5.1+cu124 (2.3.1 unavailable on cu124 index) Co-Authored-By: claude-flow --- scripts/training/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/training/Dockerfile b/scripts/training/Dockerfile index 3c6e7453f..593c55fca 100644 --- a/scripts/training/Dockerfile +++ b/scripts/training/Dockerfile @@ -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 \