# NOTE: Install PyTorch separately based on your CUDA version
# See installation instructions in README.md or https://pytorch.org/get-started/locally/
# Example: pip install torch torchvision

# Core dependencies
transformers>=4.37.0
accelerate>=0.27.0
peft>=0.8.0
datasets>=2.16.0
sentencepiece>=0.1.99
protobuf>=3.20.0

# Vision-Language Model dependencies
qwen-vl-utils>=0.0.1
pillow>=8.0,<12.0
opencv-python>=4.8.0
tiktoken>=0.5.0

# Training dependencies
deepspeed>=0.12.0
wandb>=0.16.0
tensorboard>=2.15.0
bitsandbytes>=0.41.0

# Evaluation dependencies
numpy>=1.24.0
pandas>=2.0.0
scikit-learn>=1.3.0
matplotlib>=3.7.0
seaborn>=0.12.0
tqdm>=4.65.0

# Data processing
jsonlines>=4.0.0
pyarrow>=14.0.0
scipy>=1.11.0

# Utility dependencies
einops>=0.7.0
ninja>=1.11.0
packaging>=23.0
pyyaml>=6.0
requests>=2.31.0
filelock>=3.12.0
huggingface-hub>=0.20.0

# Additional utilities
python-dotenv>=1.0.0
rich>=13.7.0
typer>=0.9.0
