huggingface_hub
torch  # Latest CUDA-enabled version
torchvision
torchaudio
transformers>=4.36.0
sentence-transformers
faiss-gpu==1.7.2 # Use GPU-optimized FAISS
accelerate>=0.26.0
bitsandbytes>=0.41.0
numpy<2.0  # Downgrade for compatibility with FAISS
gradio
tqdm
python-dotenv
langchain
langchain-community

# Machine Learning dependencies
tf-keras
tensorflow
scikit-learn
optimum
auto-gptq

# Tracking and visualization
wandb
matplotlib
seaborn
pandas

# Development dependencies
pytest
sentencepiece
protobuf