from .configuration_symphony import SymphonyConfig
from .modeling_symphony import SymphonyForConditionalGeneration

from transformers import AutoConfig, AutoModelForCausalLM, AutoModel

AutoConfig.register("symphony", SymphonyConfig)
AutoModelForCausalLM.register(SymphonyConfig, SymphonyForConditionalGeneration)
AutoModel.register(SymphonyConfig, SymphonyForConditionalGeneration)