File size: 301 Bytes
c33a4a6
 
 
 
 
 
 
1
2
3
4
5
6
7
8
from .configuration_neuronspark import NeuronSparkConfig
from .modeling_neuronspark import NeuronSparkForCausalLM

from transformers import AutoConfig, AutoModelForCausalLM

AutoConfig.register("neuronspark", NeuronSparkConfig)
AutoModelForCausalLM.register(NeuronSparkConfig, NeuronSparkForCausalLM)