torch == 1.11.0+cu113
transformers
accelerate
datasets
sentencepiece != 0.1.92
protobuf
sacrebleu >= 1.4.12
py7zr
tqdm
scikit-learn
miditoolkit
torchmetrics
wandb
