PereLluis13
/

document_index_blink

riccorl commited on Oct 31, 2023

Commit

c0b01dc

•

1 Parent(s): 90e4118

Update config.yaml (#1)

- Update config.yaml (2ab572849c9c2712f7cf205c5e4ba194c38d4fb2)

Co-authored-by: Riccardo Orlando <riccorl@users.noreply.huggingface.co>

Files changed (1) hide show

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