examples/research_projects/rag/distributed_pytorch_retriever.py/RagPyTorchDistributedRetriever/__init__
class RagPyTorchDistributedRetriever:    def __init__(self, config, question_encoder_tokenizer, generator_tokenizer, index=None):
        super().__init__(
            config,
            question_encoder_tokenizer=question_encoder_tokenizer,
            generator_tokenizer=generator_tokenizer,
            index=index,
            init_retrieval=False,
        )
        self.process_group = None