| 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 | |