Update requirements.txt

requirements.txt

@@ -1,24 +1,23 @@
-# Core dependencies
 huggingface_hub
-torch  # Will get latest CUDA-enabled version
+torch  # Latest CUDA-enabled version
 torchvision
 torchaudio
-transformers
+transformers>=4.36.0
 sentence-transformers
-faiss-gpu  # Changed from faiss-cpu for T4
-accelerate  # Added for T4
-bitsandbytes  # Added for T4 quantization
-numpy
+faiss-gpu==1.7.3  # Use GPU-optimized FAISS
+accelerate>=0.26.0
+bitsandbytes>=0.41.0
+numpy<2.0  # Downgrade for compatibility with FAISS
 gradio
 tqdm
 python-dotenv
 
 # Machine Learning dependencies
-tf-keras  # backwards-compatible keras
+tf-keras
 tensorflow
 scikit-learn
-optimum  # Added for T4 optimization
-auto-gptq  # Added for T4 quantization
+optimum
+auto-gptq
 
 # Tracking and visualization
 wandb
@@ -28,10 +27,5 @@ pandas
 
 # Development dependencies
 pytest
-sentencepiece  # Added for tokenization
+sentencepiece
 protobuf
-
-# Optional: Add minimum versions only where critical
-#accelerate>=0.26.0  # This version is required for device_map
-#transformers>=4.36.0  # For Gemma support
-#bitsandbytes>=0.41.0  # For stable 8-bit quantization