Hugging Face Transformers 在信任确认前写入自定义生成代码
Hugging Face Transformers downloads custom generation code before trust consent
AI 导读
Hugging Face Transformers 4.49.0 至 5.8.1 存在漏洞，调用 GenerativePreTrainedModel.load_custom_generate() 时会在用户确认 trust_remote_code 之前就抓取并缓存远程 Python 模块文件。攻击者控制的 custom_generate/generate.py 代码即使用户拒绝信任提示，也会被写入 ~/.cache/huggingface/modules 目录。执行环节仍受信任检查限制，但文件写入不可逆且可跨会话留存，可能造成磁盘上持久化的未授权文件，以及缓存冲突导致攻击者代码在后续受信任的模型加载中被执行。
推荐理由
公告说明了写入发生在信任确认之前，使用 Transformers 加载自定义生成代码的团队可据此检查本地缓存是否残留未授权文件。