首页
/ LM-Evaluation-Harness项目中的Tokenizer远程代码信任问题解析

LM-Evaluation-Harness项目中的Tokenizer远程代码信任问题解析

2025-05-26 23:31:44作者:温玫谨Lighthearted

在评估开源语言模型时,研究人员经常使用LM-Evaluation-Harness这一强大的评估框架。近期,有开发者在评估internlm2_5-1_8b-chat模型时遇到了一个典型的技术问题,这个问题涉及到Hugging Face Transformers库中的安全机制。

当使用本地推理服务器(vllm)加载internlm2_5-1_8b-chat模型的tokenizer时,系统会提示需要用户确认是否信任远程代码。这是因为该模型的tokenizer实现包含自定义代码,而Transformers库出于安全考虑默认不自动执行这些代码。在交互式环境中,系统会等待用户输入(y/N),但在非交互式环境(如脚本或批处理作业)中,这会导致EOFError异常。

问题的核心在于LM-Evaluation-Harness的api_models.py文件中加载tokenizer的方式。当前实现没有考虑传递trust_remote_code参数,而这个参数对于加载包含自定义代码的模型是必要的。特别是在使用API模式进行评估时,这个问题会中断整个评估流程。

技术解决方案相对直接:需要在加载tokenizer时显式传递trust_remote_code参数。这个参数应该与模型的其他参数保持一致,如果用户在模型参数中指定了trust_remote_code=True,那么加载tokenizer时也应该采用相同的设置。

这个问题反映了在模型评估工具开发中需要考虑的几个重要方面:

  1. 安全性:Hugging Face的信任机制是为了防止潜在恶意代码的执行
  2. 自动化:评估工具需要适应非交互式环境
  3. 一致性:模型加载和tokenizer加载的参数应该保持一致

对于开发者来说,理解这个问题有助于更好地处理类似包含自定义组件的模型评估场景。同时,这也提醒我们在构建模型评估流水线时,需要充分考虑各种运行环境和安全限制。

该问题的修复已经通过PR提交,允许用户通过model_args传递trust_remote_code参数,从而解决了自动化评估中的这一障碍。这个改进使得LM-Evaluation-Harness能够更顺畅地评估那些包含自定义tokenizer实现的模型,进一步提升了框架的兼容性和实用性。

登录后查看全文
热门项目推荐
相关项目推荐

项目优选

收起
docsdocs
暂无描述
Markdown
827
5.48 K
kernelkernel
openEuler内核是openEuler操作系统的核心,既是系统性能与稳定性的基石,也是连接处理器、设备与服务的桥梁。
C
494
515
ops-nnops-nn
本项目是CANN提供的神经网络类计算算子库,实现网络在NPU上加速计算。
C++
783
1.57 K
pytorchpytorch
作为 Ascend for PyTorch 社区的核心组件,TorchNPU 是昇腾专为 PyTorch 打造的深度学习适配插件,使 PyTorch 框架能够直接调用昇腾 NPU,为开发者提供昇腾 AI 处理器的超强算力。
Python
800
1.14 K
ops-transformerops-transformer
本项目是CANN提供的transformer类大模型算子库,实现网络在NPU上加速计算。
C++
970
2.28 K
kernelkernel
deepin linux kernel
C
32
16
AscendNPU-IRAscendNPU-IR
AscendNPU-IR是基于MLIR(Multi-Level Intermediate Representation)构建的,面向昇腾亲和算子编译时使用的中间表示,提供昇腾完备表达能力,通过编译优化提升昇腾AI处理器计算效率,支持通过生态框架使能昇腾AI处理器与深度调优
C++
480
312
jiuwenswarmjiuwenswarm
JiuwenSwarm 是一款基于openJiuwen开发的智能AI Agent,它能够将大语言模型的强大能力,通过你日常使用的各类通讯应用,直接延伸至你的指尖。
Python
3.01 K
766
cannbot-skillscannbot-skills
CANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体,本仓库为其提供可复用的 Skills 模块。
Markdown
1.26 K
808
cann-learning-hubcann-learning-hub
CANN 学习中心仓,支持在线互动运行、边学边练,提供教程、示例与优化方案,一站式助力昇腾开发者快速上手。
Jupyter Notebook
647
284