thak123's picture
Update requirements.txt
80266c4
raw
history blame contribute delete
132 Bytes
nltk
datasets==1.6.2
torch==1.8.1
transformers==4.8.2
pytorch-lightning==1.4.9
tokenizers==0.10.3
numpy==1.21.2
scikit-learn==0.24.1