Challenger-1 / __init__.py
MaxiiMin's picture
Update __init__.py
06f7c57 verified
raw
history blame contribute delete
294 Bytes
# __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)