fix serialization

Author: HydrogenSulfate

deepmd/pd/utils/serialization.py

@@ -88,8 +88,8 @@ def deserialize_to_file(model_file: str, data: dict) -> None:
         model.forward_lower,
         full_graph=True,
         input_spec=[
-            InputSpec([1, -1, 3], dtype="float64", name="coord"),
-            InputSpec([1, -1], dtype="int32", name="atype"),
+            InputSpec([1, -1, 3], dtype="float64", name="extended_coord"),
+            InputSpec([1, -1], dtype="int32", name="extended_atype"),
             InputSpec([1, -1, -1], dtype="int32", name="nlist"),
             None,
             None,