python(五)rag学习一:04离线模型使用
·
1、代码如下:
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# from modelscope import snapshot_download
#
# model_dir = snapshot_download("BAAI/bge-large-zh-v1.5", cache_dir="D:\LLM\Local_model")
from sentence_transformers import SentenceTransformer
model_path = r"D:\LLM\Local_model\BAAI\bge-large-zh-v1___5" # 替换为实际路径
model = SentenceTransformer(model_path)
sentences = [
"你好",
'你'
]
# 生成向量(默认返回numpy数组)
embeddings = model.encode(sentences)
print(embeddings)
print(embeddings.shape) # 输出: (2, 1024) (维度取决于模型)
2、运行结果如下:
更多推荐


所有评论(0)