be3a39d
1
2
3
4
5
6
7
from transformers import AutoConfig, AutoModelForCausalLM from .configuration_spt import SPTConfig from .modeling_spt import SPTForCausalLM AutoConfig.register("spt", SPTConfig) AutoModelForCausalLM.register(SPTConfig, SPTForCausalLM)
