orion / requirements.txt
andreslu's picture
Update requirements.txt
f770c44
raw
history blame
171 Bytes
--extra-index-url https://download.pytorch.org/whl/cu113
torch
ngram==3.3.2
nltk==3.6.2
rouge-score==0.0.4
tqdm==4.49.0
sklearn==0.0
setuptools==52.0.0
transformers==4.9.1