SYMPHONY-ASR / __init__.py
speechisalluneed's picture
Update __init__.py
3cbab94 verified
raw
history blame contribute delete
383 Bytes
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)