torch>=1.9.0
pytorch-lightning
transformers
datasets
evaluate
peft
