quangtqv/bi_encoder_tool_learning_25_7_2024_element_1_tools_rag

Uso directo (Transformadores de oraciones)

Primero, instala la biblioteca de Sentence Transformers:

pip install -U sentence-transformers

Luego, puedes cargar este modelo y ejecutar la inferencia.

from sentence_transformers import SentenceTransformer

# Descargar desde el Hub de 🤗
model = SentenceTransformer("quangtqv/bi_encoder_tool_learning_25_7_2024_element_1_tools_rag")

# Ejecutar inferencia
sentences = [
    'The weather is lovely today.',
    "It's so sunny outside!",
    'He drove to the stadium.',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 768]

# Obtener las puntuaciones de similitud para las incrustaciones
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]