| fix: 修复多个关键bug和安全隐患
Co-authored-by: 王姜奔<wangjiangben@huawei.com>
# message auto-generated for no-merge-commit merge:
!2340 merge master into master
fix: 修复多个关键bug和安全隐患
Created-by: wangjiangben
Commit-by: 王姜奔
Merged-by: ascend-robot
Description: ## 修复内容
本次提交修复了MindSpeed-MM代码仓库中的多个关键bug和安全隐患:
### 1. 修复共享内存资源泄漏问题
- 文件: mindspeed_mm/data/data_utils/utils.py
- 问题: 共享内存创建后未调用 unlink() 释放,导致内存泄漏
- 修复: 添加 try-finally 块确保资源正确释放
### 2. 修复除零风险问题
- 文件: mindspeed_mm/models/vlm_model.py
- 问题: 损失计算时 token_nums 可能为零导致除零错误
- 修复: 使用 torch.clamp 和零值检查防止除零
### 3. 修复代码注入安全隐患
- 文件: mindspeed_mm/tasks/rl/soragrpo/utils/fsdp_util.py
- 问题: 使用 eval() 函数存在代码注入风险
- 修复: 使用安全的字符串解析方式处理分数
### 4. 改进异常处理
- 文件: mindspeed_mm/data/data_utils/func_utils/mm_plugin.py
- 问题: 异常捕获过于宽泛,影响调试
- 修复: 细化异常类型,添加错误日志
## 测试
- ✅ 代码已通过本地测试
- ✅ 不影响现有功能
- ✅ 提高了代码安全性和稳定性
## 修改统计
- 修改文件: 4个
- 新增代码: 48行
- 删除代码: 22行
See merge request: Ascend/MindSpeed-MM!2340 | 1 个月前 |