#22955 [Diffusion] Fix ModelOpt B200 CI artifact coverage
作者 BBuf · 合并时间 2026-04-17 23:33
修复 ModelOpt B200 diffusion CI 覆盖,优化权重文件选择和 artifact 保存。
建议工程团队精读 `transformer_load_utils.py` 中的 `_prefer_mixed_safetensors_files` 函数,理解其设计权衡:在遇到混合和非混合文件共存时,优先选择混合版本以避免重复张量名问题。同时,关注测试 artifact 保存机制,确保在 CI 中正确配置环境变量以利用此功能。
参与讨论