已合并
修复 qwen25vl_7B mbridge 特性报错的问题 #2906
bonjour_rzq1998创建于 27 天前
修复 qwen25vl_7B mbridge 特性报错的问题 #2906
已合并
共 1 个文件变更+6-1
| @@ -11,8 +11,13 @@ except ImportError as exc: | |||
| 11 | exc_name = getattr(exc, "name", "") or "" | 11 | exc_name = getattr(exc, "name", "") or "" |
| 12 | exc_msg = str(exc) | 12 | exc_msg = str(exc) |
| 13 | 13 | ||
| 14 | + known_optional_deps = ( | ||
| 15 | + "mindspeed.te", | ||
| 16 | + "fla_npu", | ||
| 17 | + ) | ||
| 18 | + | ||
| 14 | if not ( | 19 | if not ( |
| 15 | - exc_name.startswith("mindspeed.te") | 20 | + exc_name.startswith(known_optional_deps) |
| 16 | or "mindspeed.te.pytorch.utils" in exc_msg | 21 | or "mindspeed.te.pytorch.utils" in exc_msg |
| 17 | or "get_tensor_model_parallel_group_if_none" in exc_msg | 22 | or "get_tensor_model_parallel_group_if_none" in exc_msg |
| 18 | ): | 23 | ): |