如何在 Vespa 中正确集成 ONNX 模型(使用 pyvespa)

夏枫吖_4097

夏枫吖_4097

2026-05-29

306人浏览

原创

本文详解在 pyvespa 中部署 onnx 模型时常见的路径引用错误,重点说明为何不能在应用包中直接引用宿主机或 docker 卷中的 onnx 文件,并提供可落地的解决方案:将模型文件嵌入 zip 包、正确配置 schema 与 model 定义、确保资源分配充足。

本文详解在 pyvespa 中部署 onnx 模型时常见的路径引用错误,重点说明为何不能在应用包中直接引用宿主机或 docker 卷中的 onnx 文件,并提供可落地的解决方案:将模型文件嵌入 zip 包、正确配置 schema 与 model 定义、确保资源分配充足。

在使用 pyvespa 将 ONNX 模型集成至 Vespa 向量数据库时,一个高频报错是:

json.decoder.JSONDecodeError: Expecting value: line 1 column 1 (char 0)

该错误表面看是 JSON 解析失败,但根源往往在于 ONNX 模型文件路径配置不当。Vespa 的应用包(.zip)在部署时会被完整上传并解压到容器内运行;它不支持从外部挂载卷(如 Docker volume)或宿主机绝对路径中动态加载模型文件。也就是说,以下写法是无效的:

# ❌ 错误示例:引用本地/卷路径(部署时不可达)
self.app_package.schema.add_model(
    Model(
        name="finance_onnx",
        model_file="/path/on/host/model.onnx",  # ← Vespa 容器内根本不存在此路径!
        input="input.1",
        output="output.1"
    )
)

✅ 正确做法是:将 ONNX 文件作为资源纳入应用包,并使用相对路径引用。具体步骤如下:

当贝AI
当贝AI

一款AI工具,主要用于免登录体验DeepSeek满血版,适合需要提升相关任务效率的用户。

下载
  1. 将 .onnx 文件置于项目目录下(如 models/finance_model.onnx)
  2. 在 app_package.schema.add_model() 中使用相对于 schema.sd 的路径(即包内路径)
  3. 确保 pyvespa 构建 ZIP 时包含该文件(默认会自动包含同目录及子目录下的二进制资源)

示例代码(修正后):

from vespa.package import ApplicationPackage, Schema, Document, Field, RankProfile, Model

def create_finance_app(self):
    self.app_package = ApplicationPackage(name="finance")

    # 定义文档结构(含 embedding 字段)
    self.app_package.schema = Schema(
        name="finance_doc",
        document=Document(
            fields=[
                Field(name="title", type="string"),
                Field(name="embedding", type="tensor<float>(x[1024])"),
            ]
        )
    )

    # 添加 ONNX 模型(路径为包内相对路径)
    self.app_package.schema.add_model(
        Model(
            name="finance_onnx",
            model_file="models/finance_model.onnx",  # ✅ 必须存在于项目目录中,且会被打包进 .zip
            input="input.1",
            output="output.1"
        )
    )

    # 添加 rank profile,调用模型生成 embedding 或打分
    self.app_package.schema.add_rank_profile(
        RankProfile(
            name="default",
            first_phase="closeness(field, embedding)",
            # 若需用 ONNX 做 query encoding,可配合 query pipeline + function
            functions=[
                Function(
                    name="query_embedding",
                    expression="onnx(model, \"finance_onnx\", {\"input.1\": query(query_tensor)})"
                )
            ],
            inputs=[("query(query_tensor)", "tensor<float>(x[1024])")]
        )
    )

    self.app_package.schema.mode = "streaming"
    return self.app_package</float></float>

? 关键注意事项:

  • ✅ model_file 必须是相对于 application/ 根目录的路径(如 models/xxx.onnx),且该文件必须物理存在于你的 Python 项目目录中;
  • ✅ 部署前请确认 pyvespa 构建的 ZIP 包已包含该 ONNX 文件(可通过 unzip -l app.zip 验证);
  • ⚠️ Docker 环境下需为 Vespa 容器分配足够内存(ONNX 推理需额外资源),建议至少 --memory=4g;
  • ⚠️ ONNX 模型输入/输出张量名、维度必须与 Vespa schema 和 query tensor 定义严格一致,否则 runtime 报错;
  • ? 若仍报 JSON 解析错误,请检查 deploy_app() 中是否误传了空/非法 JSON(如未正确初始化 Vespa() 实例或 deploy() 参数异常)。

总结:Vespa 的模型部署是“静态打包”范式——所有依赖(包括 ONNX、词典、配置)必须内嵌于应用包。理解并遵守这一约束,是成功集成机器学习模型的第一步。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1531

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

3604

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1549

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

20557

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2547

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2607

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1063

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

576

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2023

5

热门下载

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

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133万人学习