RAG
Collection
the best collection of RAG model, like embedding, ranker etc. • 1 item • Updated • 1
How to use Amu/tao-8k with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("Amu/tao-8k")
sentences = [
"那是 個快樂的人",
"那是 條快樂的狗",
"那是 個非常幸福的人",
"今天是晴天"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4]A try for emebdding model:
The method is the same as the stella-v2, I just extend the length of the context on tao.(I found if you want to use the fully-8k context, you maybe need to convert the model to float32).
Now I'm working on the tao-v2, It will have a different sturcture.
I will release tao-v2 as fast as I can.
Thank you to the open source community.