hooman650
/

bge-large-en-v1.5-onnx-o4

Feature Extraction

text-embeddings-inference

Model card Files Files and versions Community

hooman650 commited on Nov 22, 2023

Commit

8f45b92

·

1 Parent(s): 58177cc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,5 +29,5 @@ model.to("cuda")
 pairs = ["pandas usually live in the jungles"]
 with torch.no_grad():
     inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
-    logits = model(**inputs, return_dict=True).logits
 ```

 pairs = ["pandas usually live in the jungles"]
 with torch.no_grad():
     inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
+    logits = model(**inputs)[0][:, 0]
 ```