AIServ/models/downloadV2Model.py

# 注意：请把该文件放到文件根目录运行
# 下载或更新本地的模型
import os
from sentence_transformers import SentenceTransformer

def download_and_save_model():
    # 1.创建项目目录
    model_path = "./models/bin/all-MiniLM-L6-v2"
    os.makedirs(os.path.dirname(model_path), exist_ok=True)
    print("正在下载模型...")
    # 2.从Hugging Face下载模型
    model = SentenceTransformer('all-MiniLM-L6-v2')
    print("正在保存模型到本地...")
    # 3.保存到指定路径
    model.save(model_path)
    print(f"模型已保存到: {model_path}")
    return model

if __name__ == "__main__":
    download_and_save_model()