mkurman's picture
Upload 4 files
655f713 verified
Raw
History Blame Contribute Delete
356 Bytes
"""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)