torch>=1.9.0
transformers>=4.20.0
peft>=0.3.0
accelerate>=0.20.0
datasets>=2.0.0
tokenizers>=0.13.0
huggingface-hub>=0.15.0
click>=8.0.0
pyyaml>=6.0
numpy>=1.21.0
tqdm>=4.64.0
psutil>=5.8.0
scikit-learn>=1.0.0

[apple]
torch[mps]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
flake8>=5.0.0
mypy>=1.0.0

[gpu]
torch[cuda]
accelerate[cuda]
