-
Notifications
You must be signed in to change notification settings - Fork 0
/
requirements.txt
30 lines (28 loc) · 791 Bytes
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# local package
-e .
# external requirements
click~=7.1.2
flake8
python-dotenv>=0.5.1
pandas~=1.3.1
scispacy==0.4.0
https://s3-us-west-2.amazonaws.com/ai2-s2-scispacy/releases/v0.4.0/en_core_sci_scibert-0.4.0.tar.gz
https://s3-us-west-2.amazonaws.com/ai2-s2-scispacy/releases/v0.4.0/en_ner_bc5cdr_md-0.4.0.tar.gz
https://s3-us-west-2.amazonaws.com/ai2-s2-scispacy/releases/v0.4.0/en_core_sci_sm-0.4.0.tar.gz
https://s3-us-west-2.amazonaws.com/ai2-s2-scispacy/releases/v0.4.0/en_core_sci_lg-0.4.0.tar.gz
# huggingface_hub==0.4.0
setuptools~=52.0.0
numpy~=1.20.3
tqdm~=4.64.0
rank_bm25~=0.2.1
transformers~=4.9.2
torch~=1.9.0
scikit-learn~=0.24.2
keybert==0.6.0
gensim==4.2.0
redis[hiredis]==4.3.5
pytorch_lightning==1.7.2
dotmap==1.3.30
ir-measures==0.3.1
torchmetrics==0.11.4
wandb~=0.15.8