SYMPHONY / __init__.py
speechisalluneed's picture
Upload folder using huggingface_hub
46475d4 verified
from .configuration_fastslm import FastSLMConfig
from .modeling_fastslm import FastSLMForConditionalGeneration # FastALMForCausalLM
from transformers import AutoConfig, AutoModelForCausalLM
AutoConfig.register("fastslm", FastSLMConfig)
AutoModelForCausalLM.register(FastSLMConfig, FastSLMForConditionalGeneration)