rm unused vars

deepmodeling · Jan 30, 2024 · ce3116b · ce3116b
1 parent 73b588d
commit ce3116b
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/deepmd/pt/model/task/ener.py b/deepmd/pt/model/task/ener.py
@@ -272,8 +272,6 @@ def forward(
         """
         xx = descriptor
         nf, nloc, nd = xx.shape
-        dtype = descriptor.dtype
-        device = env.DEVICE
         # NOTICE in tests/pt/test_model.py
         # it happens that the user directly access the data memeber self.bias_atom_e
         # and set it to a wrong shape!