Relik
English
relik-cie-small / config.yaml
relik-ie's picture
Update config.yaml
071fd21 verified
raw
history blame
927 Bytes
_target_: relik.inference.annotator.Relik
retriever:
span:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: relik-ie/encoder-e5-base-v2-wikipedia
triplet:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: relik-ie/encoder-e5-small-v2-wikipedia-relations
index:
span:
_target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
name_or_path: relik-ie/encoder-e5-base-v2-wikipedia-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-cie-wikipedia
use_nme: true
task: BOTH
metadata_fields: []
top_k: 100
window_size: 64
window_stride: 32