PyTorch
bert
mrdupont commited on
Commit
443bd00
·
verified ·
1 Parent(s): 8f4121b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -16,8 +16,8 @@ mols = [
16
  ]
17
 
18
 
19
- tokenizer = AutoTokenizer.from_pretrained("UdS-LSV/da4mt-mtr-60")
20
- model = AutoModel.from_pretrained("UdS-LSV/da4mt-mtr-60")
21
 
22
  inputs = tokenizer(mols, add_special_tokens=True, truncation=True, max_length=128, padding="max_length", return_tensors="pt")
23
  embeddings = model(**inputs).last_hidden_state[:, 0, :]
 
16
  ]
17
 
18
 
19
+ tokenizer = AutoTokenizer.from_pretrained("UdS-LSV/da4mt-mlm-30")
20
+ model = AutoModel.from_pretrained("UdS-LSV/da4mt-mlm-30")
21
 
22
  inputs = tokenizer(mols, add_special_tokens=True, truncation=True, max_length=128, padding="max_length", return_tensors="pt")
23
  embeddings = model(**inputs).last_hidden_state[:, 0, :]