from transformers import AutoConfig

from .configuration_laneformer import LaneformerConfig
from .modeling_laneformer import LaneformerForCausalLM, LaneformerModel

LaneformerConfig.register_for_auto_class()
LaneformerModel.register_for_auto_class("AutoModel")
LaneformerForCausalLM.register_for_auto_class("AutoModelForCausalLM")
AutoConfig.register("laneformer", LaneformerConfig)

__all__ = ["LaneformerConfig", "LaneformerModel", "LaneformerForCausalLM"]