from .configuration_fastslm import FastSLMConfig
from .modeling_fastslm import FastSLMForConditionalGeneration # FastALMForCausalLM

from transformers import AutoConfig, AutoModelForCausalLM

AutoConfig.register("fastslm", FastSLMConfig)
AutoModelForCausalLM.register(FastSLMConfig, FastSLMForConditionalGeneration)