from transformers import BertConfig


class DenseRetrievalConfig(BertConfig):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.auto_map = {
            "AutoConfig": "configuration_denseretrieval.DenseRetrievalConfig",
            "AutoModel":  "modeling_denseretrieval.DenseRetrievalModel",
        }