Clone electric field tensor correctly

Author: Neutrino155

mace/calculators/mace.py

@@ -471,9 +471,8 @@ def calculate(self, atoms=None, properties=None, system_changes=all_changes):
             compute_polarization = True
             compute_becs = True
             compute_polarizability = True
-            batch_base["electric_field"] = torch.tensor(
-                self.electric_field, dtype=next(self.models[0].parameters()).dtype
-            )
+            batch_base["electric_field"] = self.electric_field.detach().clone()
+            
 
         ret_tensors = self._create_result_tensors(
             self.model_type, self.num_models, len(atoms)

mace/modules/extensions.py

@@ -352,11 +352,7 @@ def forward(
 
         # Setting electric field
         if electric_field is not None:
-            electric_field = (
-                torch.tensor(electric_field, dtype=vectors.dtype)
-                .view(-1, 3)
-                .requires_grad_(True)
-            )  # [num_graphs, 3]
+            electric_field = electric_field.detach().clone().view(-1, 3).requires_grad_(True) # [num_graphs, 3]
         else:
             electric_field = (
                 data["electric_field"].reshape(-1, 3).requires_grad_(True)