relik-ie's picture
Update config.yaml
ba269b3 verified
raw
history blame contribute delete
622 Bytes
_target_: relik.inference.annotator.Relik
retriever:
span:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: relik-ie/encoder-e5-base-v2-wikipedia
index:
span:
_target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
name_or_path: relik-ie/encoder-e5-base-v2-wikipedia-index
reader:
_target_: relik.reader.pytorch_modules.span.RelikReaderForSpanExtraction
transformer_model: PereLluis13/relik-reader-deberta-large-wikipedia-aida-full-interleave-robust
task: SPAN
metadata_fields: []
top_k: 100
window_size: 32
window_stride: 16
use_nme: true