Fix PyTorch/TorchVision compatibility: use exact versions 2.6.0/0.21.0

Dockerfile

@@ -22,8 +22,8 @@ WORKDIR /app
 # Copy requirements
 COPY requirements.txt .
 
-# Install PyTorch 2.6+ with CUDA support to fix CVE-2025-32434
-RUN pip install torch>=2.6.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
+# Install PyTorch 2.6.0 with compatible TorchVision 0.21.0 to fix CVE-2025-32434
+RUN pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu121
 
 # Skip llama-cpp-python to avoid compilation - using transformers instead
 

requirements.txt

@@ -1,7 +1,7 @@
 fastapi==0.104.1
 uvicorn[standard]==0.24.0
 pydantic==2.5.0
-torch>=2.6.0
+torch==2.6.0
 transformers>=4.35.0
 accelerate>=0.24.0
 bitsandbytes>=0.41.0