__init__.py

from transformers import AutoConfig, AutoModel

from .modeling_rqa import RQAModelConfig, RQAModelHF

__all__ = ["RQAModelConfig", "RQAModelHF"]

try:
    AutoConfig.register("rqa_v2_2", RQAModelConfig)
except ValueError:
    pass

try:
    AutoModel.register(RQAModelConfig, RQAModelHF)
except ValueError:
    pass