File size: 254 Bytes
f2de080
 
 
4fd51ac
f2de080
 
 
1
2
3
4
5
6
7
from transformers import AutoConfig, AutoModel

from .configuration import MetaLATTEConfig
from .modeling_metalatte import MultitaskProteinModel

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