fix: Upgrade torch via pip instead of new base image

- Use known-working base image (2.4.0)
- Upgrade torch to 2.6+ via pip
- Previous base image was too slow to pull

components/runpod_trainer/Dockerfile

@@ -1,12 +1,13 @@
-FROM runpod/pytorch:1.0.3-cu1290-torch260-ubuntu2204
+FROM runpod/pytorch:2.4.0-py3.11-cuda12.4.1-devel-ubuntu22.04
 
 WORKDIR /app
 
 # Copy requirements first for better caching
 COPY requirements.txt /app/requirements.txt
 
-# Install dependencies (torch 2.8 in base image satisfies CVE-2025-32434)
+# Upgrade torch to 2.6+ for CVE-2025-32434 compliance, then install deps
 RUN pip install --upgrade pip && \
+    pip install --no-cache-dir "torch>=2.6.0" torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 && \
     pip install --no-cache-dir -r requirements.txt
 
 # Copy handler and data