File size: 792 Bytes
37822e8
 
8197b11
626eca0
6d5df04
 
1555662
 
 
 
626eca0
 
 
 
 
 
 
 
 
 
 
 
 
37822e8
 
 
6d5df04
37822e8
 
f97bdf6
37822e8
1555662
626eca0
 
 
 
 
 
 
 
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
31
32
33
34
35
36
37
38
39
40
41
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.1.0
transformers[sentencepiece]>=4.33,<4.34
rich>=13.0.0,<14.0.0
scikit-learn
overrides
art==6.1
pprintpp==0.4.0
colorama==0.4.6

#------- Optional dependencies -------

# train
lightning>=2.0,<2.1
hydra-core>=1.3,<1.4
hydra_colorlog
wandb>=0.15,<0.16
datasets>=2.13,<2.15

# faiss
faiss-cpu==1.7.4 # needed by: faiss

# serve
fastapi>=0.104,<0.105 # needed by: serve, ray
uvicorn[standard]==0.23.2 # needed by: serve, ray
gunicorn==21.2.0 # needed by: serve, ray
spacy>=3.7,<3.8 # needed by: serve, ray
streamlit>=1.28,<1.29 # needed by: serve, ray
streamlit_extras>=0.3,<0.4 # needed by: serve, ray
pyvis # needed by: serve, ray
ray[serve]>=2.7,<=2.8 # needed by: ray

# retriever

# reader

# dev
pre-commit
black[d]
isort