fix dtype for inference model

deepmodeling · Jan 8, 2025 · f82aa99 · f82aa99
1 parent 4801b2a
commit f82aa99
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 3 deletions.
diff --git a/deepmd/pd/utils/serialization.py b/deepmd/pd/utils/serialization.py
@@ -88,7 +88,7 @@ def deserialize_to_file(model_file: str, data: dict) -> None:
         full_graph=True,
         input_spec=[
             InputSpec([1, -1, 3], dtype="float64", name="coord"),
-            InputSpec([1, -1], dtype="int64", name="atype"),
+            InputSpec([1, -1], dtype="int32", name="atype"),
             InputSpec([1, -1, -1], dtype="int32", name="nlist"),
             None,
             None,

diff --git a/source/lmp/plugin/CMakeLists.txt b/source/lmp/plugin/CMakeLists.txt
@@ -10,7 +10,7 @@ if(DEFINED LAMMPS_SOURCE_ROOT OR DEFINED LAMMPS_VERSION)
       lammps_download
       GIT_REPOSITORY https://github.com/lammps/lammps
       GIT_TAG ${LAMMPS_VERSION})
-    message(STATUS "STARTING DOWNLOAD LAMMPS TO" ${LAMMPS_SOURCE_ROOT})
+    message(STATUS "STARTING DOWNLOAD LAMMPS TO: " ${LAMMPS_SOURCE_ROOT})
     FetchContent_MakeAvailable(lammps_download)
     set(LAMMPS_SOURCE_ROOT ${lammps_download_SOURCE_DIR})
   endif()

diff --git a/source/tests/infer/deeppot_sea.forward_lower.json b/source/tests/infer/deeppot_sea.forward_lower.json
diff --git a/source/tests/infer/deeppot_sea.forward_lower.pdiparams b/source/tests/infer/deeppot_sea.forward_lower.pdiparams
diff --git a/source/tests/infer/deeppot_sea.pdiparams b/source/tests/infer/deeppot_sea.pdiparams