Update README.md

README.md

@@ -10,7 +10,7 @@ tags:
 - taylor-m1
 ---
 
-# constructai/taylor-m1-38m-128
+# constructai/taylor-m1-22m-128
 
 ---
 
@@ -23,7 +23,7 @@ The model produces **384-dimensional** embeddings (CLS token) optimized for cosi
 
 # Training details
 
-* ~38.3M parameters
+* ~22 417 920M parameters
 
 * **Max sequence length: 128 tokens**
 
@@ -66,7 +66,7 @@ from taylor_model import load_taylor_model, embed_texts
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model, tokenizer, _ = load_taylor_model("constructai/taylor-m1-38m-128", device=device)
+model, tokenizer, _ = load_taylor_model("constructai/taylor-m1-22m-128", device=device)
 
 texts = ["What is a neural network?", "How to make pizza at home?"]
 embeddings = embed_texts(model, tokenizer, texts, device=device)
@@ -80,7 +80,7 @@ from taylor_model import load_taylor_model, embed_texts
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model, tokenizer, _ = load_taylor_model("constructai/taylor-m1-38m-128", device=device)
+model, tokenizer, _ = load_taylor_model("constructai/taylor-m1-22m-128", device=device)
 
 queries = [
     "What is a neural network?",