src/transformers/models/rag/retrieval_rag.py/CustomHFIndex/__init__
class CustomHFIndex:    def __init__(self, vector_size: int, dataset, index_path=None):
        super().__init__(vector_size, dataset, index_initialized=index_path is None)
        self.index_path = index_path