PereLluis13
/

document_index_rebel

riccorl commited on Oct 31, 2023

Commit

a6935ab

•

1 Parent(s): 5a0329d

Update config.yaml (#1)

- Update config.yaml (136bdde925e799600689b5dcc0c295e95b23f4cd)

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/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