requirements.txt

# Extra index for CUDA-enabled PyTorch
--extra-index-url https://download.pytorch.org/whl/cu113

# Core dependencies
torch>=2.0.0
transformers>=4.36.0
accelerate>=0.21.0
streamlit>=1.24.0
safetensors>=0.3.1
huggingface_hub>=0.19.0

# Optimization
bitsandbytes>=0.41.0
sentencepiece>=0.1.99

# UI components
streamlit-chat>=0.1.1