Update requirements.txt

requirements.txt

@@ -1,9 +1,13 @@
 # Core dependencies
 huggingface_hub
-torch
+torch  # Will get latest CUDA-enabled version
+torchvision
+torchaudio
 transformers
 sentence-transformers
-faiss-cpu
+faiss-gpu  # Changed from faiss-cpu for T4
+accelerate  # Added for T4
+bitsandbytes  # Added for T4 quantization
 numpy
 gradio
 tqdm
@@ -13,6 +17,8 @@ python-dotenv
 tf-keras  # backwards-compatible keras
 tensorflow
 scikit-learn
+optimum  # Added for T4 optimization
+auto-gptq  # Added for T4 quantization
 
 # Tracking and visualization
 wandb
@@ -22,3 +28,10 @@ pandas
 
 # Development dependencies
 pytest
+sentencepiece  # Added for tokenization
+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