SimCSE: Simple Contrastive Learning of Sentence Embeddings
Paper • 2104.08821 • Published
Configuration Parsing Warning:Config file config.json cannot be fetched (too big)
Configuration Parsing Warning:Config file tokenizer_config.json cannot be fetched (too big)
Data
unsupervise train data is E-commerce dialogue.
Model
model is simcse.
Usage
>>> from transformers import AutoTokenizer, AutoModel
>>> model = AutoModel.from_pretrained("tuhailong/SimCSE-bert-base")
>>> tokenizer = AutoTokenizer.from_pretrained("tuhailong/SimCSE-bert-base")
>>> sentences_str_list = ["今天天气不错的","天气不错的"]
>>> inputs = tokenizer(sentences_str_list,return_tensors="pt", padding='max_length', truncation=True, max_length=32)
>>> outputs = model(**inputs)
- Downloads last month
- 22
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("tuhailong/SimCSE-bert-base", device_map="auto")