Sentence Similarity
sentence-transformers
PyTorch
ONNX
Safetensors
OpenVINO
Chinese
bert
Sentence Transformers
Instructions to use shibing624/text2vec-base-chinese with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use shibing624/text2vec-base-chinese with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("shibing624/text2vec-base-chinese") sentences = [ "那是 個快樂的人", "那是 條快樂的狗", "那是 個非常幸福的人", "今天是晴天" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Inference
- Notebooks
- Google Colab
- Kaggle
怎么启用GPU
#7
by dolphin1999 - opened
debug日志显示使用了CPU, 怎么开启GPU呢?
2023-05-18 03:12:48.433 | DEBUG | text2vec.sentence_model:init:74 - Use device: cpu
CUDA_VISIBLE_DEVICES=0 python ...
shibing624 changed discussion status to closed