requirements.txt

# Core dependencies for AI4Bharat IndicTrans2 project
torch>=2.0.0
transformers>=4.30.0
IndicTransToolkit>=1.0.0

# Optional GPU acceleration (recommended for better performance)
# Note: flash-attn requires CUDA and should be installed separately if needed:
# pip install flash-attn>=2.0.0
# flash-attn>=2.0.0

# Development and testing dependencies
pytest>=7.0.0
pytest-cov>=4.0.0

# Utility packages
numpy>=1.21.0
huggingface-hub>=0.34.0,<1.0

# For handling different data formats (if needed)
datasets>=2.10.0
sentencepiece>=0.1.97

# For logging and monitoring (optional)
wandb>=0.15.0
tensorboard>=2.10.0

#For UI and demo purposes (optional)
gradio>=4.0.0