Update config.yaml
Browse files- config.yaml +2 -2
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 |
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: PereLluis13/relik-reader-deberta-large-wikipedia-aida-full-interleave-robust
|
|
|
2 |
retriever:
|
3 |
span:
|
4 |
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
|
5 |
+
question_encoder: relik-ie/encoder-e5-base-v2-wikipedia
|
6 |
index:
|
7 |
span:
|
8 |
_target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
|
9 |
+
name_or_path: relik-ie/encoder-e5-base-v2-wikipedia-index
|
10 |
reader:
|
11 |
_target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
|
12 |
transformer_model: PereLluis13/relik-reader-deberta-large-wikipedia-aida-full-interleave-robust
|