Transformers 模型集成合同
huggingface ·
Hugging Face 推出了模型集成合同(MICs),旨在在代码合并到主分支之前捕获 Transformers 库中的回归问题。该方法解决了大规模 GPU 测试因运行速度过慢而无法在每个拉取请求中执行的问题,从而避免了诸如 T5 注意力缺陷等错误长期存在的情况。MICs 利用“夹具”(fixtures),即经过大幅缩减的真实模型检查点版本,这些版本保留了所有层类型,但所需的计算资源大大减少。每个合同都包含一个使用路径(如文本生成或训练步骤),并将输出结果与经过人工审查的基线进行对比,以确保一致性。团队开发了一种四步方法来缩小模型规模,包括保留预处理文件,并通过随机初始化或有针对性的过拟合来验证权重。一款由大语言模型辅助的命令行工具可帮助起草这些合同,尽管批准基线期望仍需人工审核。虽然目前该系统在持续集成(CI)中不具阻塞性,但其目标是通过捕捉标准前向传播可能遗漏的细微架构错误来提高可靠性。
huggingface