Update config.yaml
Browse files- config.yaml +3 -3
config.yaml
CHANGED
@@ -2,11 +2,11 @@ _target_: relik.inference.annotator.Relik
|
|
2 |
retriever:
|
3 |
span:
|
4 |
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
|
5 |
-
question_encoder:
|
6 |
-
document_index:
|
7 |
reader:
|
8 |
_target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
|
9 |
-
transformer_model:
|
10 |
task: SPAN
|
11 |
metadata_fields: []
|
12 |
top_k: 100
|
|
|
2 |
retriever:
|
3 |
span:
|
4 |
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
|
5 |
+
question_encoder: sapienzanlp/relik-retriever-e5-base-v2-aida-blink-encoder
|
6 |
+
document_index: sapienzanlp/relik-retriever-e5-base-v2-aida-blink-index
|
7 |
reader:
|
8 |
_target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
|
9 |
+
transformer_model: sapienzanlp/relik-reader-deberta-v3-large-aida
|
10 |
task: SPAN
|
11 |
metadata_fields: []
|
12 |
top_k: 100
|