diff --git a/deepmd/pt/model/model/atomic_model.py b/deepmd/pt/model/model/atomic_model.py
index 47fd463fc9..9720bfa57d 100644
--- a/deepmd/pt/model/model/atomic_model.py
+++ b/deepmd/pt/model/model/atomic_model.py
@@ -14,16 +14,9 @@
 from deepmd.model_format import (
     FittingOutputDef,
 )
-from deepmd.pt.model.task import (
-    Fitting,
-)
 
 
 class AtomicModel(ABC):
-    @abstractmethod
-    def get_fitting_net(self) -> Fitting:
-        raise NotImplementedError
-
     @abstractmethod
     def get_fitting_output_def(self) -> FittingOutputDef:
         raise NotImplementedError
diff --git a/deepmd/pt/model/model/dp_atomic_model.py b/deepmd/pt/model/model/dp_atomic_model.py
index ffeeeda660..a0f9b25765 100644
--- a/deepmd/pt/model/model/dp_atomic_model.py
+++ b/deepmd/pt/model/model/dp_atomic_model.py
@@ -128,12 +128,6 @@ def __init__(
                     self.descriptor.dim_out, self.ntypes - 1, self.descriptor.dim_emb
                 )
 
-    def get_fitting_net(self) -> Fitting:
-        """Get the fitting net."""
-        return (
-            self.fitting_net if self.fitting_net is not None else self.coord_denoise_net
-        )
-
     def get_fitting_output_def(self) -> FittingOutputDef:
         """Get the output def of the fitting net."""
         return (