Update deepmd/dpmodel/descriptor/repformers.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: Jinzhe Zeng <[email protected]>

deepmd/dpmodel/descriptor/repformers.py

@@ -516,14 +516,13 @@ def serialize(self):
             # variables
             "g2_embd": self.g2_embd.serialize(),
             "repformer_layers": [layer.serialize() for layer in self.layers],
-            "env_mat": EnvMat(self.rcut, self.rcut_smth).serialize(),
+            "env_mat": self.env_mat.serialize(),
             "@variables": {
                 "davg": to_numpy_array(self["davg"]),
                 "dstd": to_numpy_array(self["dstd"]),
             },
         }
 
-
 # translated by GPT and modified
 def get_residual(
     _dim: int,