__init__.py

from .modeling_rqa import RQAModelConfig, RQAModelHF

__all__ = ["RQAModelConfig", "RQAModelHF"]

from transformers import AutoConfig, AutoModel

AutoConfig.register("rqa", RQAModelConfig)
AutoModel.register(RQAModelConfig, RQAModelHF)