configs/index.yaml

trainer:
  logger: false
  callbacks:
  - class_path: lightning_ir.IndexCallback
    init_args:
      index_dir: ./index
      index_config:
        class_path: FaissIVFPQIndexConfig
        init_args:
          num_centroids: 262144
          num_subquantizers: 16
          n_bits: 8
model:
  class_path: lightning_ir.BiEncoderModule
  init_args:
    model_name_or_path: webis/bert-bi-encoder
data:
  class_path: lightning_ir.LightningIRDataModule
  init_args:
    num_workers: 1
    inference_batch_size: 256
    inference_datasets:
      - class_path: DocDataset
        init_args:
          doc_dataset: msmarco-passage