Spaces:
Build error
Build error
update some files
Browse files- install_env.sh +1 -1
- 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
|
| 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 \
|