Update README.md

README.md

@@ -6,9 +6,9 @@ tags:
 - peptide
 ---
 
-# Model Card for Mistral-Peptide-v1-16M (Mistral for peptide)
+# Model Card for Mistral-Peptide-v1-15M (Mistral for peptide)
 
-The Mistral-Peptide-v1-16M Large Language Model (LLM) is a pretrained generative peptide molecule model with 15.2M parameters. 
+The Mistral-Peptide-v1-15M Large Language Model (LLM) is a pretrained generative peptide molecule model with 15.2M parameters. 
 It is derived from Mixtral-8x7B-v0.1 model, which was simplified for protein: the number of layers and the hidden size were reduced. 
 The model was pretrained using 863499 peptide strings. 
 
@@ -26,8 +26,8 @@ Like Mixtral-8x7B-v0.1, it is a transformer model, with the following architectu
 import torch
 from transformers import AutoTokenizer, AutoModel
 
-tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-Peptide-v1-16M", trust_remote_code=True) 
-model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Peptide-v1-16M", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-Peptide-v1-15M", trust_remote_code=True) 
+model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Peptide-v1-15M", trust_remote_code=True)
 ```
 
 ## Calculate the embedding of a protein sequence
@@ -48,7 +48,7 @@ Ensure you are utilizing a stable version of Transformers, 4.34.0 or newer.
 
 ## Notice
 
-Mistral-Peptide-v1-16M is a pretrained base model for peptide.
+Mistral-Peptide-v1-15M is a pretrained base model for peptide.
 
 ## Contact