base_IIXIV / fla /models /samba /__init__.py
mainline777's picture
Duplicate from silx-ai/Quasar-Preview
41865df
Raw
History Blame Contribute Delete
474 Bytes
from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
from fla.models.samba.configuration_samba import SambaConfig
from fla.models.samba.modeling_samba import SambaForCausalLM, SambaModel
AutoConfig.register(SambaConfig.model_type, SambaConfig, exist_ok=True)
AutoModel.register(SambaConfig, SambaModel, exist_ok=True)
AutoModelForCausalLM.register(SambaConfig, SambaForCausalLM, exist_ok=True)
__all__ = ['SambaConfig', 'SambaForCausalLM', 'SambaModel']