Relik
English
riccorl commited on
Commit
8e0d408
1 Parent(s): 648cd1d

Update config.yaml

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