SimCSE: Simple Contrastive Learning of Sentence Embeddings
Paper β’ 2104.08821 β’ Published
How to use cyclone/simcse-chinese-roberta-wwm-ext with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("feature-extraction", model="cyclone/simcse-chinese-roberta-wwm-ext") # Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("cyclone/simcse-chinese-roberta-wwm-ext")
model = AutoModel.from_pretrained("cyclone/simcse-chinese-roberta-wwm-ext")YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
This model provides simplified Chinese sentence embeddings encoding based on Simple Contrastive Learning. The pretrained model(Chinese RoBERTa WWM Ext) is used for token encoding.
Please use SentenceTransformer to load the model.
from sentence_transformers import SentenceTransformer
encoder = SentenceTransformer('cyclone/simcse-chinese-roberta-wwm-ext')