RaphaelMourad
/

Mistral-DNA-v0.1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

RaphaelMourad commited on Jan 25

Commit

3495c1e

•

1 Parent(s): c8ed588

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ Like Mistral-7B-v0.1, it is a transformer model, with the following architecture
 import torch
 from transformers import AutoTokenizer, AutoModel
-tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-DNA-v0.1", trust_remote_code=True)
 model = AutoModel.from_pretrained("RaphaelMourad/Mistral-DNA-v0.1", trust_remote_code=True)
 ```

 import torch
 from transformers import AutoTokenizer, AutoModel
+tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-DNA-v0.1", trust_remote_code=True) # Same as DNABERT2
 model = AutoModel.from_pretrained("RaphaelMourad/Mistral-DNA-v0.1", trust_remote_code=True)
 ```