del ener model precison

deepmodeling · Feb 28, 2024 · 1fa1afc · 1fa1afc
1 parent b1ad4bc
commit 1fa1afc
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/deepmd/pt/model/model/ener_model.py b/deepmd/pt/model/model/ener_model.py
@@ -30,9 +30,6 @@ def forward(
         aparam: Optional[torch.Tensor] = None,
         do_atomic_virial: bool = False,
     ) -> Dict[str, torch.Tensor]:
-        coord = coord.to(env.GLOBAL_PT_FLOAT_PRECISION)
-        if box is not None:
-            box = box.to(env.GLOBAL_PT_FLOAT_PRECISION)
         model_ret = self.forward_common(
             coord,
             atype,