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