Skip to content

Commit c7b643b

Browse files
gshtrasxuebwang-amd
authored andcommitted
[Bugfix] Fix for the import error from vllm-project#24588 (vllm-project#25481)
Signed-off-by: Gregory Shtrasberg <[email protected]> Signed-off-by: xuebwang-amd <[email protected]>
1 parent 56ad4a8 commit c7b643b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/layers/fused_moe/gpt_oss_triton_kernels_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from triton_kernels.routing import (RoutingData, routing,
2424
routing_from_bitmatrix)
2525
from triton_kernels.tensor import Bitmatrix
26-
except (ModuleNotFoundError, AttributeError) as e:
26+
except (AttributeError, ImportError) as e:
2727
logger.error(
2828
"Failed to import Triton kernels. Please make sure your triton "
2929
"version is compatible. Error: %s", e)

0 commit comments

Comments
 (0)