fix: upgrade to CUDA 12.4 + torch>=2.6 for CVE-2025-32434 compat

Dockerfile

@@ -1,11 +1,11 @@
-FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
+FROM nvidia/cuda:12.4.0-cudnn-runtime-ubuntu22.04
 
 RUN apt-get update && apt-get install -y \
     python3 python3-pip git \
     && rm -rf /var/lib/apt/lists/*
 
 RUN pip3 install --no-cache-dir \
-    torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
+    torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 && \
     pip3 install --no-cache-dir \
     flask flask-cors numpy "transformers>=4.40,<5" accelerate sentencepiece protobuf \
     lightning omegaconf safetensors diffusers ftfy regex