from transformers import AutoConfig, AutoModel
from .configuration import MetaLATTEConfig
from .modeling_metalatte import MultitaskProteinModel
AutoConfig.register("metalatte", MetaLATTEConfig)
AutoModel.register(MetaLATTEConfig, MultitaskProteinModel) | 
