用pipeline加载embedding时,报错:KeyError: ‘TextGenerationPipeline is not in the pipelines registry group embedding. Please make sure the correct version of ModelScope library is used.’
KeyError: 'TextGenerationPipeline不在管道注册表组嵌入中。请确保使用了正确版本的ModelScope库。
from modelscope.pipelines import pipeline
embedding_model_name = "BAAI/bge-large-zh-v1.5"
embedding_pipeline = pipeline(task="embedding", model=embedding_model_name)
embeddings = embedding_pipeline(texts) # 生成嵌入向量
然后我换成了
from sentence_transformers import SentenceTransformer
embedding_model = SentenceTransformer('BAAI/bge-large-zh-v1.5')
embeddings = embedding_model.encode(texts, normalize_embeddings=True)
这种方式加载,就OK了