Relik
English
riccorl commited on
Commit
a6b4690
1 Parent(s): 1f4b689

Update config.yaml

Browse files
Files changed (1) hide show
  1. 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: 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
 
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