ModelScope 임베딩

ModelScope 임베딩

이 노트북에서는 LlamaIndex에서 ModelScope 임베딩을 사용하는 방법을 보여드려요. 자세한 내용은 ModelScope 사이트를 참고하면 돼요.

출처: 문서

본문

Colab에서 이 노트북을 여는 경우라면 LlamaIndex 🦙와 modelscope가 필요해요.

!pip install llama-index-embeddings-modelscope

기본 사용법 (Basic Usage)

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)

배치 임베딩 생성 (Generate Batch Embedding)

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)

더 알아보기 (Learn more)

• ModelScope 사이트