PereLluis13
/

document_index_blink

Model card Files Files and versions Community

riccorl commited on Oct 31, 2023

Commit

2ab5728

•

1 Parent(s): 90e4118

Update config.yaml

Files changed (1) hide show

config.yaml +2 -3

config.yaml CHANGED Viewed

@@ -1,11 +1,10 @@
 _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex
 metadata_fields:
-  _target_: omegaconf.listconfig.ListConfig
 separator: ' <def> '
 documents:
   _target_: relik.retriever.indexers.document.DocumentStore
 embeddings:
   _target_: torch.Tensor
-name_or_dir: /home/huguetcabot/alby-re/golden-retriever/experiments/blink/retriever/document_index
-device_in_init: cuda
 precision: fp32

 _target_: relik.retriever.indexers.inmemory.InMemoryDocumentIndex
 metadata_fields:
+- definition
 separator: ' <def> '
 documents:
   _target_: relik.retriever.indexers.document.DocumentStore
 embeddings:
   _target_: torch.Tensor
+name_or_path: /home/huguetcabot/alby-re/golden-retriever/experiments/blink/retriever/document_index
 precision: fp32