forked from IntelLabs/fastRAG
-
Notifications
You must be signed in to change notification settings - Fork 0
/
fact_check_pipeline.yaml
46 lines (46 loc) · 969 Bytes
/
fact_check_pipeline.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
components:
- name: Store
params:
host: localhost
index: index-name
port: 80
type: ElasticsearchDocumentStore
- name: Retriever
params:
document_store: Store
top_k: 10
type: BM25Retriever
- name: RagRetriever
params:
document_store: null
query_embedding_model: rag_query_embedding_model
passage_embedding_model: facebook/dpr-ctx_encoder-single-nq-base
type: DensePassageRetriever
- name: Reranker
params:
batch_size: 32
model_name_or_path: cross-encoder/ms-marco-MiniLM-L-12-v2
top_k: 10
use_gpu: true
type: SentenceTransformersRanker
- name: Reader
params:
model_name_or_path: fact_checking_model_name_or_path
retriever: RagRetriever
top_k: 1
min_length: 1
use_gpu: true
type: RAGenerator
pipelines:
- name: my_pipeline
nodes:
- inputs:
- Query
name: Retriever
- inputs:
- Retriever
name: Reranker
- inputs:
- Reranker
name: Reader
version: 1.12.2