ModelScope 嵌入

在本笔记本中，我们将展示如何在LlamaIndex中使用ModelScope嵌入。请查看ModelScope网站。

如果您在 Colab 上打开这个笔记本，您需要安装 LlamaIndex 🦙 和 modelscope。

!pip install llama-index-embeddings-modelscope

基本用法

import sys
from llama_index.embeddings.modelscope.base import ModelScopeEmbedding

model = ModelScopeEmbedding(
    model_name="iic/nlp_gte_sentence-embedding_chinese-base",
    model_revision="master",
)

rsp = model.get_query_embedding("Hello, who are you?")
print(rsp)

rsp = model.get_text_embedding("Hello, who are you?")
print(rsp)

生成批量嵌入

from llama_index.embeddings.modelscope.base import ModelScopeEmbedding

model = ModelScopeEmbedding(
    model_name="iic/nlp_gte_sentence-embedding_chinese-base",
    model_revision="master",
)

rsp = model.get_text_embedding_batch(
    ["Hello, who are you?", "I am a student."]
)
print(rsp)