Relik
English
relik-ie commited on
Commit
aefb0ec
1 Parent(s): e88394f

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +2 -2
config.yaml CHANGED
@@ -2,14 +2,14 @@ _target_: relik.inference.annotator.Relik
2
  retriever:
3
  span:
4
  _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
5
- question_encoder: relik-ie/encoder-e5-small-v2-wikipedia-matryoshka
6
  triplet:
7
  _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
8
  question_encoder: relik-ie/encoder-e5-small-v2-wikipedia-relations
9
  index:
10
  span:
11
  _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
12
- name_or_path: relik-ie/index-e5-small-v2-wikipedia-matryoshka
13
  triplet:
14
  _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
15
  name_or_path: relik-ie/encoder-e5-small-v2-wikipedia-relations-index
 
2
  retriever:
3
  span:
4
  _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
5
+ question_encoder: relik-ie/encoder-e5-base-v2-wikipedia-matryoshka
6
  triplet:
7
  _target_: relik.retriever.pytorch_modules.model.GoldenRetriever
8
  question_encoder: relik-ie/encoder-e5-small-v2-wikipedia-relations
9
  index:
10
  span:
11
  _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
12
+ name_or_path: relik-ie/index-e5-base-v2-wikipedia-matryoshka
13
  triplet:
14
  _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained
15
  name_or_path: relik-ie/encoder-e5-small-v2-wikipedia-relations-index