xpmir
/

cross-encoder-bert-base-infoNCE

Text Classification

sequence-classification

text-embeddings-inference

Model card Files Files and versions

Metrics Training metrics Community

VictorMorand commited on 20 days ago

Commit

fb1db57

·

verified ·

1 Parent(s): f0c87e6

Upload README.md with huggingface_hub

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ Quick Start:
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-tokenizer = AutoTokenizer.from_pretrained("cross-encoder-bert-base-infoNCE")
 model = AutoModelForSequenceClassification.from_pretrained("xpmir/cross-encoder-bert-base-infoNCE")
 features = tokenizer("What is experimaestro ?", "Experimaestro is a powerful framework for ML experiments management...", padding=True, truncation=True, return_tensors="pt")

 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
+tokenizer = AutoTokenizer.from_pretrained("xpmir/cross-encoder-bert-base-infoNCE")
 model = AutoModelForSequenceClassification.from_pretrained("xpmir/cross-encoder-bert-base-infoNCE")
 features = tokenizer("What is experimaestro ?", "Experimaestro is a powerful framework for ML experiments management...", padding=True, truncation=True, return_tensors="pt")