Update model name after org rename
Browse files
README.md
CHANGED
@@ -25,8 +25,8 @@ def mean_pooling(token_embeddings, mask):
|
|
25 |
return sentence_embeddings
|
26 |
|
27 |
# Remove pooling layer
|
28 |
-
model = AutoModel.from_pretrained("
|
29 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
30 |
|
31 |
sentences = [
|
32 |
"Where was Marie Curie born?", # Query
|
|
|
25 |
return sentence_embeddings
|
26 |
|
27 |
# Remove pooling layer
|
28 |
+
model = AutoModel.from_pretrained("CarperAI/carptriever-1", add_pooling_layer=False)
|
29 |
+
tokenizer = AutoTokenizer.from_pretrained("CarperAI/carptriever-1")
|
30 |
|
31 |
sentences = [
|
32 |
"Where was Marie Curie born?", # Query
|