summarization / requirements.txt
gagan3012's picture
split added
b1cba3a
raw
history blame
229 Bytes
numpy==1.21.1
datasets==1.10.2
pytorch_lightning==1.3.5
transformers==4.9.0
torch==1.9.0
dagshub==0.1.6
pandas==1.2.4
rouge_score
pyyaml
dvc
mlflow
wandb
# external requirements
click
coverage
awscli
flake8
python-dotenv>=0.5.1