Update README.md
Browse files
README.md
CHANGED
|
@@ -10,7 +10,7 @@ tags:
|
|
| 10 |
- taylor-m1
|
| 11 |
---
|
| 12 |
|
| 13 |
-
# constructai/taylor-m1
|
| 14 |
|
| 15 |
---
|
| 16 |
|
|
@@ -66,7 +66,7 @@ from taylor_model import load_taylor_model, embed_texts
|
|
| 66 |
import torch
|
| 67 |
|
| 68 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 69 |
-
model, tokenizer, _ = load_taylor_model("constructai/taylor-m1
|
| 70 |
|
| 71 |
texts = ["What is a neural network?", "How to make pizza at home?"]
|
| 72 |
embeddings = embed_texts(model, tokenizer, texts, device=device)
|
|
@@ -80,7 +80,7 @@ from taylor_model import load_taylor_model, embed_texts
|
|
| 80 |
import torch
|
| 81 |
|
| 82 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 83 |
-
model, tokenizer, _ = load_taylor_model("constructai/taylor-m1
|
| 84 |
|
| 85 |
queries = [
|
| 86 |
"What is a neural network?",
|
|
|
|
| 10 |
- taylor-m1
|
| 11 |
---
|
| 12 |
|
| 13 |
+
# constructai/taylor-m1
|
| 14 |
|
| 15 |
---
|
| 16 |
|
|
|
|
| 66 |
import torch
|
| 67 |
|
| 68 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 69 |
+
model, tokenizer, _ = load_taylor_model("constructai/taylor-m1", device=device)
|
| 70 |
|
| 71 |
texts = ["What is a neural network?", "How to make pizza at home?"]
|
| 72 |
embeddings = embed_texts(model, tokenizer, texts, device=device)
|
|
|
|
| 80 |
import torch
|
| 81 |
|
| 82 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 83 |
+
model, tokenizer, _ = load_taylor_model("constructai/taylor-m1", device=device)
|
| 84 |
|
| 85 |
queries = [
|
| 86 |
"What is a neural network?",
|