requirements.txt

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

# Core dependencies
torch
transformers>=4.31.0
accelerate>=0.21.0
streamlit>=1.24.0
safetensors>=0.3.1

# Optimization
bitsandbytes>=0.41.0
sentencepiece>=0.1.99

# UI components
streamlit-chat>=0.1.1