yrshi commited on
Commit
9cd28e9
·
1 Parent(s): 81fc00b

update some files

Browse files
Files changed (2) hide show
  1. install_env.sh +1 -1
  2. retrieval_launch.sh +1 -1
install_env.sh CHANGED
@@ -7,4 +7,4 @@ pip install transformers datasets pyserini
7
 
8
  conda install -c pytorch -c nvidia faiss-gpu=1.8.0 -y
9
 
10
- pip install uvicorn fastapi huggingface_hub
 
7
 
8
  conda install -c pytorch -c nvidia faiss-gpu=1.8.0 -y
9
 
10
+ pip install uvicorn fastapi huggingface_hub pydantic
retrieval_launch.sh CHANGED
@@ -5,7 +5,7 @@ corpus_file=$file_path/wiki-18.jsonl
5
  retriever=intfloat/e5-base-v2
6
 
7
  export CUDA_VISIBLE_DEVICES="1,3"
8
- python search_r1/search/retrieval_server.py --index_path $index_file \
9
  --corpus_path $corpus_file \
10
  --topk 3 \
11
  --retriever_model $retriever \
 
5
  retriever=intfloat/e5-base-v2
6
 
7
  export CUDA_VISIBLE_DEVICES="1,3"
8
+ python retrieval_server.py --index_path $index_file \
9
  --corpus_path $corpus_file \
10
  --topk 3 \
11
  --retriever_model $retriever \