Update requirements.txt

requirements.txt

@@ -1,14 +1,14 @@
---extra-index-url https://download.pytorch.org/whl/cu118
+--extra-index-url https://download.pytorch.org/whl/cu121
 
-# Core compute (CUDA 11.8, good for Tesla T4)
-torch==2.2.2+cu118
+# Core compute (CUDA 12.x -> use cu121 wheels)
+torch==2.3.1+cu121
 
 # NLP stack
 transformers==4.43.3
-tokenizers==0.19.1       # compatible with transformers 4.43.x; avoids tokenizer.json issue
+tokenizers==0.19.1       # compatible with transformers 4.43.x
 accelerate==0.33.0
 safetensors==0.4.4
-bitsandbytes==0.43.1
+bitsandbytes==0.43.1     # works with CUDA 12.x
 sentencepiece==0.1.99
 huggingface_hub==0.24.6
 regex==2024.7.24