Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,7 @@ Here is a simple code for inference:
|
|
39 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
40 |
import torch
|
41 |
|
42 |
-
model_name = 'plant-nucleotide-transformer'
|
43 |
# load model and tokenizer
|
44 |
model = AutoModelForMaskedLM.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|
45 |
tokenizer = AutoTokenizer.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|
|
|
39 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
40 |
import torch
|
41 |
|
42 |
+
model_name = 'plant-nucleotide-transformer-BPE'
|
43 |
# load model and tokenizer
|
44 |
model = AutoModelForMaskedLM.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|
45 |
tokenizer = AutoTokenizer.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
|