torch>=2.0.0
transformers>=4.35.0
peft>=0.7.0
bitsandbytes>=0.41.0
accelerate>=0.24.0
numpy>=1.21.0
pandas>=1.3.0
pyyaml>=6.0
tqdm>=4.64.0
click>=8.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
isort>=5.12.0
flake8>=6.0.0
mypy>=1.0.0
pre-commit>=3.0.0

[full]
flash-attn>=2.0.0
wandb>=0.15.0
torch[cuda]>=2.0.0
triton>=2.0.0
jupyter>=1.0.0
matplotlib>=3.5.0
seaborn>=0.11.0
plotly>=5.0.0
scikit-learn>=1.0.0

[gpu]
flash-attn>=2.0.0
torch[cuda]>=2.0.0
triton>=2.0.0
