0

0

Python 语义搜索的 Sentence-Transformers 部署

冷漠man

冷漠man

发布时间:2026-02-13 23:01:03

|

329人浏览过

|

来源于php中文网

原创

根本原因是默认float32加载大模型致显存溢出;应改用float16、禁用padding预分配、分批处理;相似度不准因归一化重复或模型不一致;部署需单例加载、启用fast tokenizer;windows下慎用torch.compile,优先onnx+cpu推理。

python 语义搜索的 sentence-transformers 部署

模型加载慢、显存爆掉,model.encode() 卡住不动

根本原因不是代码写错了,而是默认用 torch.float32 加载 500MB+ 的模型(比如 sentence-transformers/all-MiniLM-L6-v2),显存吃满后触发 CPU 回退或 OOM。别急着换显卡,先改加载方式。

  • 强制用 torch.float16:加参数 device='cuda',torch_dtype=torch.float16(注意不是 fp16=True,那个只影响推理,不缩模型权重)
  • 禁用 tokenizer 的 padding 预分配:传 tokenizer_kwargs={'padding': False, 'truncation': True},否则 batch 大时内存翻倍
  • 小批量处理:别一次性 encode(list_of_10000_texts),拆成每批 64 或 128 条,用 batch_size=64 参数控制

搜索结果不准,cos_sim() 返回的相似度全在 0.8 以上

不是模型不行,是文本预处理和向量归一化没对齐。Sentence-Transformers 的 encode() 默认已做 L2 归一化,但很多人自己算余弦相似度时又手动归一,导致数值失真。

  • 直接用库里的 util.cos_sim(),别手写 np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b))
  • query 和 doc 必须用同一个模型 encode,别 query 用 all-MiniLM-L6-v2,doc 用 multi-qa-MiniLM-L6-dot-v1 —— 向量空间不兼容,相似度无意义
  • 中文文本务必确认 tokenizer 是否支持:比如 all-MiniLM-L6-v2 是多语言但非中文特化,遇到“微信”“支付宝”这类词嵌入偏弱,可换 paraphrase-multilingual-MiniLM-L12-v2

Docker 部署后 API 响应延迟高,curl 测出来首字节要 2s+

核心瓶颈常在模型首次加载 —— 每个请求都走一遍 SentenceTransformer(...),而不是复用实例。FastAPI/Flask 里别把 model 当局部变量。

Lyrics Generator
Lyrics Generator

免费人工智能歌词生成器和人工智能歌曲作家

下载
  • 全局单例加载:在模块顶层 or app.on_event("startup") 里初始化 model,确保只 load 一次
  • 关闭 tokenizer 的 slow tokenizers:加 use_fast=True 到 model 初始化参数,否则 Hugging Face 的 slow tokenizer 会拖慢首条请求 300ms+
  • 验证是否真缓存:启动后立刻发一条请求,再 time curl http://localhost:8000/search?q=test,第二次响应应 ≤200ms;如果还是慢,检查是否误写了 model = SentenceTransformer(...) 在 route 函数里

Windows 下 torch.compile() 报错或无效

torch.compile() 在 Windows 上目前仅支持 CUDA 12.1+ 且需 nightly 版本 PyTorch,生产环境别硬上。更稳的提速方案是提前编译推理图 + 关闭梯度。

立即学习Python免费学习笔记(深入)”;

  • 部署前加 model.eval()torch.no_grad()(后者必须显式写,不能只靠 eval)
  • torch.jit.script(model) 编译(注意:只支持部分模型结构,all-MiniLM-L6-v2 可行;若报错就跳过)
  • Windows 用户优先用 onnxruntime:导出 ONNX 后用 CPU 推理,比原生 torch 快 1.5–2x,且无平台兼容问题

最易被忽略的是:模型路径含中文或空格时,SentenceTransformer() 在 Windows 下可能静默失败,日志里也不报错 —— 改用绝对路径并确保全是 ASCII 字符。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python Flask框架
Python Flask框架

本专题专注于 Python 轻量级 Web 框架 Flask 的学习与实战,内容涵盖路由与视图、模板渲染、表单处理、数据库集成、用户认证以及RESTful API 开发。通过博客系统、任务管理工具与微服务接口等项目实战,帮助学员掌握 Flask 在快速构建小型到中型 Web 应用中的核心技能。

96

2025.08.25

Python Flask Web框架与API开发
Python Flask Web框架与API开发

本专题系统介绍 Python Flask Web框架的基础与进阶应用,包括Flask路由、请求与响应、模板渲染、表单处理、安全性加固、数据库集成(SQLAlchemy)、以及使用Flask构建 RESTful API 服务。通过多个实战项目,帮助学习者掌握使用 Flask 开发高效、可扩展的 Web 应用与 API。

73

2025.12.15

Python FastAPI异步API开发_Python怎么用FastAPI构建异步API
Python FastAPI异步API开发_Python怎么用FastAPI构建异步API

Python FastAPI 异步开发利用 async/await 关键字,通过定义异步视图函数、使用异步数据库库 (如 databases)、异步 HTTP 客户端 (如 httpx),并结合后台任务队列(如 Celery)和异步依赖项,实现高效的 I/O 密集型 API,显著提升吞吐量和响应速度,尤其适用于处理数据库查询、网络请求等耗时操作,无需阻塞主线程。

27

2025.12.22

Python 微服务架构与 FastAPI 框架
Python 微服务架构与 FastAPI 框架

本专题系统讲解 Python 微服务架构设计与 FastAPI 框架应用,涵盖 FastAPI 的快速开发、路由与依赖注入、数据模型验证、API 文档自动生成、OAuth2 与 JWT 身份验证、异步支持、部署与扩展等。通过实际案例,帮助学习者掌握 使用 FastAPI 构建高效、可扩展的微服务应用,提高服务响应速度与系统可维护性。

156

2026.02.06

curl_exec
curl_exec

curl_exec函数是PHP cURL函数列表中的一种,它的功能是执行一个cURL会话。给大家总结了一下php curl_exec函数的一些用法实例,这个函数应该在初始化一个cURL会话并且全部的选项都被设置后被调用。他的返回值成功时返回TRUE, 或者在失败时返回FALSE。

450

2023.06.14

linux常见下载安装工具
linux常见下载安装工具

linux常见下载安装工具有APT、YUM、DNF、Snapcraft、Flatpak、AppImage、Wget、Curl等。想了解更多linux常见下载安装工具相关内容,可以阅读本专题下面的文章。

180

2023.10.30

css中的padding属性作用
css中的padding属性作用

在CSS中,padding属性用于设置元素的内边距。想了解更多padding的相关内容,可以阅读本专题下面的文章。

141

2023.12.07

常见的编码方式
常见的编码方式

常见的编码方式有ASCII编码、Unicode编码、UTF-8编码、UTF-16编码、GBK编码等。想了解更多编码方式相关内容,可以阅读本专题下面的文章。

621

2023.10.24

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

23

2026.02.13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.2万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.5万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号