relik-ie's picture
Update config.yaml
2301acb verified
raw
history blame
619 Bytes
_target_: relik.inference.annotator.Relik
retriever:
triplet:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: relik-ie/encoder-e5-small-v2-wikipedia-relations
index:
triplet:
_target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
name_or_path: relik-ie/encoder-e5-small-v2-wikipedia-relations-index
reader:
_target_: relik.reader.pytorch_modules.triplet.RelikReaderForTripletExtraction
transformer_model: relik-ie/relik-reader-deberta-v3-small-re-wikipedia
task: TRIPLET
metadata_fields: []
top_k: 30
window_size: 64
window_stride: 32