#48791 [ModelRunner V2] Enable sequence pooling for embedding and classification models
原始 PR · 作者 taneem-ibrahim · 合并时间 2026-07-29 21:30
MRV2 支持序列级 embed/classify 池化执行
值得精读。这是 MRV2 pooling 迁移的核心执行 PR,其中三个设计决策值得学习:一是用子类隔离 encoder-only 注意力并复用 AttentionGroup 机制,二是用 CPU 侧 finished_mask 避免 GPU->CPU 往返,三是测试以 Hugging Face 为独立参考而不是 V1 互相对照。建议结合 #49331、#48290 连起来看,掌握 MRV2 启用流程的完整链路。
参与讨论