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
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
You can’t perform that action at this time.
0 commit comments