sapienzanlp
/

relik-entity-linking-base

Model card Files Files and versions

riccorl commited on Jun 10, 2024

Commit

50d96b3

·

verified ·

1 Parent(s): 5030f55

Update config.yaml

Files changed (1) hide show

config.yaml +4 -1

config.yaml CHANGED Viewed

@@ -3,7 +3,10 @@ retriever:
   span:
     _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
     question_encoder: sapienzanlp/relik-retriever-e5-base-v2-aida-blink-encoder
-    document_index: sapienzanlp/relik-retriever-e5-base-v2-aida-blink-index
 reader:
   _target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
   transformer_model: sapienzanlp/relik-reader-deberta-v3-base-aida

   span:
     _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
     question_encoder: sapienzanlp/relik-retriever-e5-base-v2-aida-blink-encoder
+index:
+  span:
+    _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
+    name_or_path: sapienzanlp/relik-retriever-e5-base-v2-aida-blink-wikipedia-index
 reader:
   _target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
   transformer_model: sapienzanlp/relik-reader-deberta-v3-base-aida