Update README.md
Browse files
README.md
CHANGED
|
@@ -92,7 +92,7 @@ curl -X POST https://api.perplexity.ai/v1/contextualizedembeddings \
|
|
| 92 |
from transformers import AutoModel
|
| 93 |
|
| 94 |
model = AutoModel.from_pretrained(
|
| 95 |
-
"
|
| 96 |
trust_remote_code=True
|
| 97 |
)
|
| 98 |
texts = [
|
|
@@ -106,7 +106,7 @@ texts = [
|
|
| 106 |
embeddings = model.encode(texts) # Shape: (5, 1024)
|
| 107 |
|
| 108 |
model_ctx = AutoModel.from_pretrained(
|
| 109 |
-
"
|
| 110 |
trust_remote_code=True
|
| 111 |
)
|
| 112 |
|
|
@@ -135,7 +135,7 @@ embeddings = model_ctx.encode(doc_chunks)
|
|
| 135 |
from sentence_transformers import SentenceTransformer
|
| 136 |
|
| 137 |
model = SentenceTransformer(
|
| 138 |
-
"
|
| 139 |
trust_remote_code=True
|
| 140 |
)
|
| 141 |
|
|
@@ -150,7 +150,7 @@ texts = [
|
|
| 150 |
embeddings = model.encode(texts) # Shape: (5, 1024)
|
| 151 |
|
| 152 |
model_ctx = SentenceTransformer(
|
| 153 |
-
"
|
| 154 |
trust_remote_code=True
|
| 155 |
)
|
| 156 |
|
|
|
|
| 92 |
from transformers import AutoModel
|
| 93 |
|
| 94 |
model = AutoModel.from_pretrained(
|
| 95 |
+
"perplexity-ai/pplx-embed-1-0.6B",
|
| 96 |
trust_remote_code=True
|
| 97 |
)
|
| 98 |
texts = [
|
|
|
|
| 106 |
embeddings = model.encode(texts) # Shape: (5, 1024)
|
| 107 |
|
| 108 |
model_ctx = AutoModel.from_pretrained(
|
| 109 |
+
"perplexity-ai/pplx-embed-1-context-0.6B",
|
| 110 |
trust_remote_code=True
|
| 111 |
)
|
| 112 |
|
|
|
|
| 135 |
from sentence_transformers import SentenceTransformer
|
| 136 |
|
| 137 |
model = SentenceTransformer(
|
| 138 |
+
"perplexity-ai/pplx-embed-1-0.6B",
|
| 139 |
trust_remote_code=True
|
| 140 |
)
|
| 141 |
|
|
|
|
| 150 |
embeddings = model.encode(texts) # Shape: (5, 1024)
|
| 151 |
|
| 152 |
model_ctx = SentenceTransformer(
|
| 153 |
+
"perplexity-ai/pplx-embed-1-context-0.6B",
|
| 154 |
trust_remote_code=True
|
| 155 |
)
|
| 156 |
|