Lolalb
/

AMPLIFY_350M

Feature Extraction

Model card Files Files and versions Community

Lolalb commited on 25 days ago

Commit

b8defd9

·

verified ·

1 Parent(s): 8c37c51

Upload AMPLIFY

Files changed (1) hide show

amplify.py +1 -1

amplify.py CHANGED Viewed

@@ -279,7 +279,7 @@ class AMPLIFY(AMPLIFYPreTrainedModel):
         model.load_state_dict(state_dict)
         cfg["tokenizer"]["vocab_path"] = "/home/mila/l/lola.lebreton/AMPLIFY/conf/tokenizer/amplify_vocab.txt"
-        tokenizer = ProteinTokenizer(**cfg["tokenizer"], max_length=cfg["trainer"]["train"]["max_length"])
         return model, tokenizer
     def forward(

         model.load_state_dict(state_dict)
         cfg["tokenizer"]["vocab_path"] = "/home/mila/l/lola.lebreton/AMPLIFY/conf/tokenizer/amplify_vocab.txt"
+        tokenizer = ProteinTokenizer(**cfg["tokenizer"], model_max_length=cfg["trainer"]["train"]["max_length"])
         return model, tokenizer
     def forward(