File size: 241 Bytes
5b3f5a5
 
 
c4cb3a0
5b3f5a5
 
 
1
2
3
4
5
6
7
from transformers import AutoConfig, AutoModel

from .configuration import MetaLATTEConfig
from .model import MultitaskProteinModel

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