Brain2nd's picture
squash old LFS history
774da7f
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)