File size: 294 Bytes
9f40432 06f7c57 9f40432 |
1 2 3 4 5 6 7 8 |
# __init__.py
from .configuration_challenger import ChallengerConfig
from .modeling_challenger import ChallengerModel
from transformers import AutoConfig, AutoModelForCausalLM
AutoConfig.register("challenger", ChallengerConfig)
AutoModelForCausalLM.register(ChallengerConfig, ChallengerModel)
|