bigscience

/

sgpt-bloom-7b1-msmarco

like 42

Follow

BigScience Workshop 798

import numpy as np
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("bigscience/sgpt-bloom-7b1-msmarco")

# Our sentences we like to encode
sentences = [
    'The quick brown fox jumps over the lazy dog.',
    'Hello World!'
]

# Sentences are encoded by calling model.encode()
embeddings = model.encode(sentences, normalize_embeddings=True)

TypeError: Pooling.__init__() got an unexpected keyword argument 'pooling_mode_weightedmean_tokens'