Brain2nd's picture
Update ckpt_step2000
33236d4 verified
raw
history blame contribute delete
301 Bytes
from .configuration_neuronspark import NeuronSparkConfig
from .modeling_neuronspark import NeuronSparkForCausalLM
from transformers import AutoConfig, AutoModelForCausalLM
AutoConfig.register("neuronspark", NeuronSparkConfig)
AutoModelForCausalLM.register(NeuronSparkConfig, NeuronSparkForCausalLM)