Hugging Face's logo

ChatterjeeLab
/
MetaLATTE

Transformers
PyTorch
metalatte
climate
biology
Model card Files
xet
Community
yinuozhang commited on
Commit
b2e0ffb
·
1 Parent(s): 935f2f3

update

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -28,9 +28,10 @@ import sys
28
  from transformers import AutoTokenizer, AutoModel, AutoConfig
29
  metalatte_path = './Chatterjeelab/MetaLATTE'
30
  sys.path.insert(0, metalatte_path)
31
- from metalatte import MetaLATTEConfig, MetaLATTEModel
32
  AutoConfig.register("metalatte", MetaLATTEConfig)
33
- AutoModel.register(MetaLATTEConfig, MetaLATTEModel)
 
34
 
35
  tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t33_650M_UR50D")
36
  config = AutoConfig.from_pretrained("ChatterjeeLab/MetaLATTE")
 
28
  from transformers import AutoTokenizer, AutoModel, AutoConfig
29
  metalatte_path = './Chatterjeelab/MetaLATTE'
30
  sys.path.insert(0, metalatte_path)
31
+ from metalatte import MetaLATTEConfig, MultitaskProteinModel
32
  AutoConfig.register("metalatte", MetaLATTEConfig)
33
+ AutoModel.register(MetaLATTEConfig, MultitaskProteinModel)
34
+
35
 
36
  tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t33_650M_UR50D")
37
  config = AutoConfig.from_pretrained("ChatterjeeLab/MetaLATTE")