yilunzhang
/

all-mpnet-base-v2-onnx

Sentence Similarity

sentence-transformers

feature-extraction

Model card Files Files and versions

yilunzhang commited on Dec 19, 2023

Commit

44b10fe

·

1 Parent(s): d79586f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -45,8 +45,8 @@ def mean_pooling(model_output, attention_mask):
 sentences = ['This is an example sentence', 'Each sentence is converted']
 # Load model from HuggingFace Hub
-tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
-model = ORTModelForFeatureExtraction.from_pretrained('sentence-transformers/all-mpnet-base-v2')
 # Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')

 sentences = ['This is an example sentence', 'Each sentence is converted']
 # Load model from HuggingFace Hub
+tokenizer = AutoTokenizer.from_pretrained('yilunzhang/all-mpnet-base-v2-onnx')
+model = ORTModelForFeatureExtraction.from_pretrained('yilunzhang/all-mpnet-base-v2-onnx')
 # Tokenize sentences
 encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')