riccorl's picture
Update config.yaml
a6b4690 verified
raw
history blame
No virus
679 Bytes
_target_: relik.inference.annotator.Relik
retriever:
span:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: riccorl/retriever-relik-entity-linking-aida-wikipedia-base-question-encoder
index:
span:
_target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
name_or_dir: riccorl/retriever-relik-entity-linking-aida-wikipedia-base-index
reader:
_target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
transformer_model: riccorl/relik-reader-deberta-base-retriever-relik-entity-linking-aida-wikipedia-large-v2
task: SPAN
metadata_fields: []
top_k: 100
window_size: 32
window_stride: 16