PearlIsa's picture
Update requirements.txt
8d4809b verified
raw
history blame
764 Bytes
# Core Libraries
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
# Hugging Face and LangChain
huggingface_hub>=0.12.0 # Specify version for compatibility
langchain>=0.2.0 # Ensure latest compatible version
langchain-community>=0.2.0 # Align with updated imports
# Gradio for Interface
gradio
# Machine Learning dependencies
tf-keras
tensorflow
scikit-learn
optimum
auto-gptq
# Tracking and visualization
wandb
matplotlib
seaborn
pandas
# Development dependencies
pytest
sentencepiece
protobuf
ipython # Required for IPython display
# Utilities
python-dotenv
tqdm