torch>=1.12.0
transformers>=4.30.0
numpy>=1.23.0
scipy>=1.9.0
tqdm>=4.65.0

[all]
lm-eval>=0.4.0
datasets>=2.10.0
evaluate>=0.4.0
peft>=0.5.0
pytorch-lightning>=2.0.0
wandb>=0.15.0
tensorboard>=2.11.0

[dev]
pytest>=7.0.0
black>=22.0.0
flake8>=4.0.0
jupyter>=1.0.0

[eval]
lm-eval>=0.4.0
datasets>=2.10.0
evaluate>=0.4.0

[finetune]
peft>=0.5.0
pytorch-lightning>=2.0.0

[logging]
wandb>=0.15.0
tensorboard>=2.11.0
