FastSLM / __init__.py
speechisalluneed's picture
Upload folder using huggingface_hub
46475d4 verified
Raw
History Blame Contribute Delete
316 Bytes
from .configuration_fastslm import FastSLMConfig
from .modeling_fastslm import FastSLMForConditionalGeneration # FastALMForCausalLM
from transformers import AutoConfig, AutoModelForCausalLM
AutoConfig.register("fastslm", FastSLMConfig)
AutoModelForCausalLM.register(FastSLMConfig, FastSLMForConditionalGeneration)