relik-ie's picture
Update config.yaml
4cb425c verified
_target_: relik.inference.annotator.Relik
retriever:
span:
_target_: relik.retriever.pytorch_modules.model.GoldenSillyRetriever
documents: ['media', 'disease', 'miscellaneous', 'event', 'person', 'location', 'time', 'celestial', 'organization', 'concept']
triplet:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: relik-ie/encoder-e5-small-v2-wikipedia-relations
index:
span:
_target_: relik.retriever.indexers.base.BaseDocumentIndex
documents: ['media', 'disease', 'miscellaneous', 'event', 'person', 'location', 'time', 'celestial', 'organization', 'concept']
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-ner-wikipedia
use_nme: true
task: BOTH
metadata_fields: []
top_k: 30
window_size: 64
window_stride: 32