Skip to content

Commit d997c80

Browse files
committed
allow gpu
1 parent fdd643d commit d997c80

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/haystack/pipeline/nlp_pipeline.yaml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,14 @@ components:
77
document_store: MyQNADocumentStore
88
embedding_model: sentence-transformers/all-MiniLM-L6-v2
99
scale_score: false
10+
use_gpu: true
1011
top_k: 5
1112

1213
- name: SparseRetriever
1314
type: BM25Retriever
1415
params:
1516
document_store: DocumentStore
17+
use_gpu: true
1618
top_k: 10
1719

1820
- name: JoinAns
@@ -26,6 +28,7 @@ components:
2628
type: FARMReader
2729
params:
2830
model_name_or_path: deepset/roberta-base-squad2
31+
use_gpu: true
2932
top_k: 5
3033

3134
- name: QNADocsToAnswers

0 commit comments

Comments
 (0)