Brain2nd's picture
Add fixed __init__.py (inference cache + per-tensor mixed precision)
c33a4a6 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)