Skip to content

Commit d5c807b

Browse files
authored
Update semantic_search.yaml (#7708)
1 parent 5de7e57 commit d5c807b

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pipelines/rest_api/pipeline/semantic_search.yaml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,20 +6,20 @@ components: # define all the building-blocks for Pipeline
66
params:
77
host: localhost
88
port: 9200
9-
index: dureader_robust_query_encoder
10-
embedding_dim: 312
9+
index: dureader_robust_base_encoder
10+
embedding_dim: 768
1111
- name: Retriever
1212
type: DensePassageRetriever
1313
params:
1414
document_store: DocumentStore # params can reference other components defined in the YAML
1515
top_k: 10
16-
query_embedding_model: rocketqa-zh-nano-query-encoder
17-
passage_embedding_model: rocketqa-zh-nano-para-encoder
16+
query_embedding_model: rocketqa-zh-base-query-encoder
17+
passage_embedding_model: rocketqa-zh-base-para-encoder
1818
embed_title: False
1919
- name: Ranker # custom-name for the component; helpful for visualization & debugging
2020
type: ErnieRanker # pipelines Class name for the component
2121
params:
22-
model_name_or_path: rocketqa-nano-cross-encoder
22+
model_name_or_path: rocketqa-base-cross-encoder
2323
top_k: 3
2424
- name: TextFileConverter
2525
type: TextConverter

0 commit comments

Comments
 (0)