"""AutoClass registration helpers for ConvGPT-v2."""

from transformers import AutoConfig, AutoModelForCausalLM

from .configuration_convgpt_v2 import ConvGPTV2Config
from .modeling_convgpt_v2 import ConvGPTV2ForCausalLM

AutoConfig.register(ConvGPTV2Config.model_type, ConvGPTV2Config)
AutoModelForCausalLM.register(ConvGPTV2Config, ConvGPTV2ForCausalLM)