PereLluis13
/

document_index_rebel

Model card Files Files and versions Community

riccorl commited on Oct 31, 2023

Commit

136bdde

•

1 Parent(s): 5a0329d

Update config.yaml

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