__init__.py

from transformers import AutoConfig, AutoModel

from .configuration import MetaLATTEConfig
from .modeling_metalatte import MultitaskProteinModel

AutoConfig.register("metalatte", MetaLATTEConfig)
AutoModel.register(MetaLATTEConfig, MultitaskProteinModel)