File size: 316 Bytes
46475d4 | 1 2 3 4 5 6 7 8 | from .configuration_fastslm import FastSLMConfig
from .modeling_fastslm import FastSLMForConditionalGeneration # FastALMForCausalLM
from transformers import AutoConfig, AutoModelForCausalLM
AutoConfig.register("fastslm", FastSLMConfig)
AutoModelForCausalLM.register(FastSLMConfig, FastSLMForConditionalGeneration)
|