Relik
English
relik-cie-small / config.yaml
relik-ie's picture
Update config.yaml
aefb0ec verified
raw
history blame contribute delete
No virus
941 Bytes
_target_: relik.inference.annotator.Relik
retriever:
span:
_target_: relik.retriever.pytorch_modules.model.GoldenRetriever
question_encoder: relik-ie/encoder-e5-base-v2-wikipedia-matryoshka
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/index-e5-base-v2-wikipedia-matryoshka
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