File size: 383 Bytes
3cbab94 e5422ca 3cbab94 e5422ca 3cbab94 | 1 2 3 4 5 6 7 8 9 | 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)
|