How to use uniem/base-softmax-last-mean with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("uniem/base-softmax-last-mean") model = AutoModel.from_pretrained("uniem/base-softmax-last-mean", device_map="auto")
pip install uniem
from uniem import UniEmbedder embedder = UniEmbedder.from_pretrained('uniem/base-softmax-last-mean') embedder.encode(['Hello World!', 'ไฝ ๅฅฝ,ไธ็!'])
