PereLluis13
/

document_index_rebel

Model card Files Files and versions Community

Update config.yaml

#1

by riccorl - opened Oct 31, 2023

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

config.yaml +3 -4

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/relik_new/relik/wandb/latest-run/files/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/relik_new/relik/wandb/latest-run/files/retriever/document_index
+precision: fp32