yinuozhang commited on
Commit
4fd51ac
1 Parent(s): 87084fa

Update __init__.py

Browse files
Files changed (1) hide show
  1. __init__.py +1 -1
__init__.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import AutoConfig, AutoModel
2
 
3
  from .configuration import MetaLATTEConfig
4
- from .model import MultitaskProteinModel
5
 
6
  AutoConfig.register("metalatte", MetaLATTEConfig)
7
  AutoModel.register(MetaLATTEConfig, MultitaskProteinModel)
 
1
  from transformers import AutoConfig, AutoModel
2
 
3
  from .configuration import MetaLATTEConfig
4
+ from .modeling_metalatte import MultitaskProteinModel
5
 
6
  AutoConfig.register("metalatte", MetaLATTEConfig)
7
  AutoModel.register(MetaLATTEConfig, MultitaskProteinModel)