swulling
/

bge-reranker-base-onnx-o4

Text Classification

text-embeddings-inference

Model card Files Files and versions

swulling commited on Nov 21, 2023

Commit

6bbccaa

·

1 Parent(s): 597d3a1

update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -24,6 +24,7 @@ tokenizer = AutoTokenizer.from_pretrained('swulling/bge-reranker-base-onnx-o4')
 model = ORTModelForSequenceClassification.from_pretrained('swulling/bge-reranker-base-onnx-o4')
 model.to("cuda")
 with torch.no_grad():
     inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
     scores = model(**inputs, return_dict=True).logits.view(-1, ).float()

 model = ORTModelForSequenceClassification.from_pretrained('swulling/bge-reranker-base-onnx-o4')
 model.to("cuda")
+pairs = [['what is panda?', 'hi'], ['what is panda?', 'The giant panda (Ailuropoda melanoleuca), sometimes called a panda bear or simply panda, is a bear species endemic to China.']]
 with torch.no_grad():
     inputs = tokenizer(pairs, padding=True, truncation=True, return_tensors='pt', max_length=512)
     scores = model(**inputs, return_dict=True).logits.view(-1, ).float()