嵌入模型BAAI/bge-small-zh-v1.5下载（不用魔法）

_喵酱_

于 2025-04-05 17:22:32 发布

阅读量366

点赞数 3

文章标签：嵌入模型

本文链接：https://blog.csdn.net/qq_39208536/article/details/147013454

版权

# -*- coding:utf-8 -*-
# @Author: 喵酱
# @time: 2025 - 04 -05
# @File: miao_test.py
# desc:
import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
from transformers import AutoModel, AutoTokenizer

model_name = "BAAI/bge-small-zh-v1.5"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModel.from_pretrained(model_name)

# 保存模型到本地目录（例如 ./bge-small-zh-v1.5）
save_path = "./bge-small-zh-v1.5"
model.save_pretrained(save_path)
tokenizer.save_pretrained(save_path)
if __name__ == '__main__':
    print(f"模型已保存到 {save_path}")