Update requirements.txt

requirements.txt

@@ -1,15 +1,15 @@
 # =========================
-# Core dependencies
+# Core
 # =========================
-torch==2.5.1               # Pinned to ensure FA2 wheel availability (cu121)
-transformers>=4.45.0       # Recent enough for latest attention API
+torch==2.5.1                 # stable + manylinux/cu12 wheels available
+transformers>=4.45.0
 accelerate>=0.30.0
 sentencepiece>=0.1.99
 protobuf>=3.20.0
 huggingface_hub>=0.16.0
 
 # =========================
-# Dataset and evaluation
+# Dataset & evaluation
 # =========================
 datasets>=2.14.0
 pandas>=1.5.0
@@ -18,7 +18,7 @@ scikit-learn>=1.0.0
 scipy>=1.8.0
 
 # =========================
-# Memory & GPU monitoring
+# Monitoring
 # =========================
 psutil>=5.9.0
 pynvml>=11.5.0
@@ -30,17 +30,11 @@ matplotlib>=3.5.0
 seaborn>=0.11.0
 
 # =========================
-# Compression / Quantization
+# Quantization (optional)
 # =========================
-bitsandbytes>=0.41.0       # For INT8 / 4-bit quantization
+bitsandbytes>=0.41.0
 
 # =========================
-# UI (Gradio)
+# UI (optional)
 # =========================
 gradio>=4.0.0
-
-# =========================
-# FlashAttention 2
-# =========================
-# Installed last so torch is already present (avoids build metadata error)
-flash-attn>=2.5.6