# -*- coding: utf-8 -*-

from transformers import AutoConfig, AutoModel, AutoModelForCausalLM

from fla.models.scan.configuration_scan import SCANConfig
from fla.models.scan.modeling_scan import SCANForCausalLM, SCANModel

AutoConfig.register(SCANConfig.model_type, SCANConfig)
AutoModel.register(SCANConfig, SCANModel)
AutoModelForCausalLM.register(SCANConfig, SCANForCausalLM)


__all__ = ['SCANConfig', 'SCANForCausalLM', 'SCANModel']