diff --git a/deepmd/infer/model_devi.py b/deepmd/infer/model_devi.py
index a45f96805b..802e0ae401 100644
--- a/deepmd/infer/model_devi.py
+++ b/deepmd/infer/model_devi.py
@@ -1,5 +1,14 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
-__all_ = [
+from deepmd_utils.infer.model_devi import (
+    calc_model_devi,
+    calc_model_devi_e,
+    calc_model_devi_f,
+    calc_model_devi_v,
+    make_model_devi,
+    write_model_devi_out,
+)
+
+__all__ = [
     "make_model_devi",
     "calc_model_devi",
     "write_model_devi_out",