diff --git a/requirements.txt b/requirements.txt index 59bd6b6..8f617f6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,11 @@ tqdm>=4.45.0 numpy>=1.18.5 -pyserini==0.9.4.0 -transformers==3.4.0 +pyserini==0.16.0 +transformers==4.17.0 torch==1.5.1 torchvision==0.6.1 tensorboardX>=2.1 hanziconv>=0.3.2 nltk +faiss-cpu +zhon