efederici
/

e5-base-v2-4096

Sentence Similarity

feature-extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

efederici commited on Jun 16, 2023

Commit

c6a5295

•

1 Parent(s): 985b3fc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ input_texts = [
 ]
 tokenizer = AutoTokenizer.from_pretrained('efederici/e5-base-v2-4096')
-model = AutoModel.from_pretrained('efederici/e5-base-v2-4096')
 batch_dict = tokenizer(input_texts, max_length=512, padding=True, truncation=True, return_tensors='pt')
 outputs = model(**batch_dict)

 ]
 tokenizer = AutoTokenizer.from_pretrained('efederici/e5-base-v2-4096')
+model = AutoModel.from_pretrained('efederici/e5-base-v2-4096', trust_remote_code=True)
 batch_dict = tokenizer(input_texts, max_length=512, padding=True, truncation=True, return_tensors='pt')
 outputs = model(**batch_dict)