diff --git a/deepmd/pt/__init__.py b/deepmd/pt/__init__.py
index ab61736198..daf3d406e9 100644
--- a/deepmd/pt/__init__.py
+++ b/deepmd/pt/__init__.py
@@ -4,6 +4,11 @@
 from deepmd.pt.cxx_op import (
     ENABLE_CUSTOMIZED_OP,
 )
+from deepmd.utils.entry_point import (
+    load_entry_point,
+)
+
+load_entry_point("deepmd.pt")
 
 __all__ = [
     "ENABLE_CUSTOMIZED_OP",
diff --git a/deepmd/pt/model/__init__.py b/deepmd/pt/model/__init__.py
index 8422ac3802..6ceb116d85 100644
--- a/deepmd/pt/model/__init__.py
+++ b/deepmd/pt/model/__init__.py
@@ -1,6 +1 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
-from deepmd.utils.entry_point import (
-    load_entry_point,
-)
-
-load_entry_point("deepmd.pt")