fix bug (#10995)

zhangbo9674 · web-flow · commit fb6d8af3c868 · 2025-08-25T14:04:30.000+08:00
diff --git a/paddlenlp/transformers/moe_utils.py b/paddlenlp/transformers/moe_utils.py
@@ -16,7 +16,11 @@
 
 import numpy as np
 import paddle
-import TokenDispatcherUtils as TDU
+
+try:
+    import TokenDispatcherUtils as TDU
+except ImportError:
+    TDU = None
 
 from .fp8_utils import FP8LinearFunctionBase
 
