pytorch-lightning==1.4.9
ct-loss
torch==1.10.0
numpy==1.19.5
tqdm==4.62.2

#metrics
torchmetrics==0.4.1

# extras
fairscale==0.4.0

# huggingface
transformers==4.10.3
datasets==1.12.1

# hydra
hydra-core==1.1.1

# task-specific
rouge-score==0.0.4
sentencepiece==0.1.96
