在本 Notebook 中，我们将展示如何在 LlamaIndex 中使用 ModelScope Embeddings。请访问 ModelScope 网站。

如果你在 Colab 中打开这个 Notebook，需要安装 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)