Update requirements.txt

requirements.txt

@@ -3,11 +3,12 @@ uvicorn[standard]>=0.30
 gradio>=4.38
 requests>=2.31
 
-# vLLM & deps
-vllm>=0.5.2
-transformers>=4.43
-accelerate
+# Match vLLM ↔ Torch versions
+vllm==0.5.5
 
-# Torch with CUDA 12.1 wheels (matches our base image)
+# Use CUDA 12.1 wheels for torch 2.4.0
 --extra-index-url https://download.pytorch.org/whl/cu121
-torch==2.2.2
+torch==2.4.0
+
+transformers>=4.44
+accelerate>=0.30