Update config.yaml
Browse files- config.yaml +3 -3
config.yaml
CHANGED
@@ -2,14 +2,14 @@ _target_: relik.inference.annotator.Relik
|
|
2 |
retriever:
|
3 |
span:
|
4 |
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
|
5 |
-
question_encoder:
|
6 |
index:
|
7 |
span:
|
8 |
_target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
|
9 |
-
name_or_path:
|
10 |
reader:
|
11 |
_target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
|
12 |
-
transformer_model:
|
13 |
task: SPAN
|
14 |
metadata_fields: []
|
15 |
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 |
index:
|
7 |
span:
|
8 |
_target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
|
9 |
+
name_or_path: sapienzanlp/relik-retriever-e5-base-v2-aida-blink-wikipedia-index
|
10 |
reader:
|
11 |
_target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
|
12 |
+
transformer_model: sapienzanlp/relik-reader-deberta-v3-large-aida
|
13 |
task: SPAN
|
14 |
metadata_fields: []
|
15 |
top_k: 100
|