Update README.md

README.md

@@ -6,9 +6,9 @@ tags:
 - chemistry
 ---
 
-# Model Card for Mistral-Chem-v1-139M (Mistral for chemistry)
+# Model Card for Mistral-Chem-v1-134M (Mistral for chemistry)
 
-The Mistral-Chem-v1-139M Large Language Model (LLM) is a pretrained generative chemical molecule model with 1.9M parameters x 8 experts = 15.2M parameters. 
+The Mistral-Chem-v1-134M Large Language Model (LLM) is a pretrained generative chemical molecule model with 134M 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 10M molecule SMILES strings from the ZINC 15 database. 
 
@@ -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-Chem-v1-139M", trust_remote_code=True) 
-model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Chem-v1-139M", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("RaphaelMourad/Mistral-Chem-v1-134M", trust_remote_code=True) 
+model = AutoModel.from_pretrained("RaphaelMourad/Mistral-Chem-v1-134M", trust_remote_code=True)
 ```
 
 ## Calculate the embedding of a DNA sequence
@@ -48,7 +48,7 @@ Ensure you are utilizing a stable version of Transformers, 4.34.0 or newer.
 
 ## Notice
 
-Mistral-Chem-v1-139M is a pretrained base model for chemistry.
+Mistral-Chem-v1-134M is a pretrained base model for chemistry.
 
 ## Contact