sapienzanlp
/

relik-relation-extraction-nyt-large

Model card Files Files and versions

PereLluis13 commited on Jul 19, 2024

Commit

c388c83

·

verified ·

1 Parent(s): 88dcb76

Update config.yaml

Files changed (1) hide show

config.yaml +3 -3

config.yaml CHANGED Viewed

@@ -2,16 +2,16 @@ _target_: relik.inference.annotator.Relik
 retriever:
   triplet:
     _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
-    question_encoder: PereLluis13/relik-retriever-small-nyt-question-encoder
 index:
   triplet:
     _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
-    name_or_path: PereLluis13/relik-retriever-small-nyt-document-index
 reader:
   _target_: relik.reader.pytorch_modules.triplet.RelikReaderForTripletExtraction
   transformer_model: sapienzanlp/relik-reader-deberta-v3-large-nyt
 task: TRIPLET
 metadata_fields: []
-top_k: 100
 window_size: sentence
 window_stride: null

 retriever:
   triplet:
     _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
+    question_encoder: sapienzanlp/relik-retriever-small-nyt-question-encoder
 index:
   triplet:
     _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
+    name_or_path: sapienzanlp/relik-retriever-small-nyt-document-index
 reader:
   _target_: relik.reader.pytorch_modules.triplet.RelikReaderForTripletExtraction
   transformer_model: sapienzanlp/relik-reader-deberta-v3-large-nyt
 task: TRIPLET
 metadata_fields: []
+top_k: 24
 window_size: sentence
 window_stride: null