Update requirements.txt

requirements.txt

@@ -1,10 +1,10 @@
 ###############################################################################
-#  CUDA 12.1 runtime  +  PyTorch 2.1 wheels  (official extra index)
+# CUDA 12.2 + PyTorch 2.1                                                    #
 ###############################################################################
---extra-index-url https://download.pytorch.org/whl/cu121
+--extra-index-url https://download.pytorch.org/whl/cu122
 
-torch==2.1.0+cu121            # core
-torchvision==0.16.0+cu121     # must share the same ABI tag
+torch==2.1.0+cu122
+torchvision==0.16.0+cu122
 
 ###############################################################################
 # Transformers / training utils
@@ -39,10 +39,5 @@ huggingface_hub
 sentencepiece
 safetensors
 
-###############################################################################
-# Flash-Attention 2.5.8  ── prebuilt for
-#   • CUDA 12.2 (binary still works under CUDA ≥12.1 drivers*)
-#   • PyTorch 2.1
-#   • Python 3.10  (cp310)
-###############################################################################
-https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
+# Flash-Attention 2.5.8 compiled for CU12.2, Torch 2.1, Python 3.10
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl  :contentReference[oaicite:0]{index=0}