TensorFlow -> PyTorch

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · Jun 22, 2024 · 3a703aa · 3a703aa
1 parent 6d2c609
commit 3a703aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/pt/cxx_op.py b/deepmd/pt/cxx_op.py
@@ -50,10 +50,10 @@ def load_library(module_name: str) -> bool:
             if PT_CXX11_ABI_FLAG != pt_cxx11_abi_flag:
                 raise RuntimeError(
                     "This deepmd-kit package was compiled with "
-                    "CXX11_ABI_FLAG=%d, but TensorFlow runtime was compiled "
+                    "CXX11_ABI_FLAG=%d, but PyTorch runtime was compiled "
                     "with CXX11_ABI_FLAG=%d. These two library ABIs are "
                     "incompatible and thus an error is raised when loading %s. "
-                    "You need to rebuild deepmd-kit against this TensorFlow "
+                    "You need to rebuild deepmd-kit against this PyTorch "
                     "runtime."
                     % (
                         PT_CXX11_ABI_FLAG,