File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
pipelines/rest_api/pipeline Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -6,20 +6,20 @@ components: # define all the building-blocks for Pipeline
6
6
params :
7
7
host : localhost
8
8
port : 9200
9
- index : dureader_robust_query_encoder
10
- embedding_dim : 312
9
+ index : dureader_robust_base_encoder
10
+ embedding_dim : 768
11
11
- name : Retriever
12
12
type : DensePassageRetriever
13
13
params :
14
14
document_store : DocumentStore # params can reference other components defined in the YAML
15
15
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
18
18
embed_title : False
19
19
- name : Ranker # custom-name for the component; helpful for visualization & debugging
20
20
type : ErnieRanker # pipelines Class name for the component
21
21
params :
22
- model_name_or_path : rocketqa-nano -cross-encoder
22
+ model_name_or_path : rocketqa-base -cross-encoder
23
23
top_k : 3
24
24
- name : TextFileConverter
25
25
type : TextConverter
You can’t perform that action at this time.
0 commit comments