requirements.txt

torch
transformers==4.25.1
ruamel-yaml==0.17.21
accelerate==0.13.2
dill==0.3.6
einops==0.6.0
wandb==0.13.8
scikit-learn
pytorch-crf==0.7.2
ordered-set==4.1.0