torch>=1.10.0
transformers>=4.20.0
numpy>=1.19.0
scikit-learn>=1.0.0
nltk>=3.6.0
rouge-score>=0.1.0
datasets>=2.0.0
tqdm>=4.62.0

[all]
peft>=0.4.0
accelerate>=0.20.0
bitsandbytes>=0.40.0
optimum>=1.8.0

[dev]
pytest>=7.0.0
pytest-cov>=3.0.0
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0
mypy>=0.940

[docs]
sphinx>=4.4.0
sphinx-rtd-theme>=1.0.0
myst-parser>=0.18.0

[fine-tuning]
peft>=0.4.0
accelerate>=0.20.0
bitsandbytes>=0.40.0

[quantization]
bitsandbytes>=0.40.0
optimum>=1.8.0
