fix typo

Author: caic99

deepmd/pt/model/descriptor/dpa3.py

@@ -474,13 +474,13 @@ def forward(
             The smooth switch function. shape: nf x nloc x nnei
 
         """
-        parrallel_mode = comm_dict is not None
+        parallel_mode = comm_dict is not None
         # cast the input to internal precsion
         extended_coord = extended_coord.to(dtype=self.prec)
         nframes, nloc, nnei = nlist.shape
         nall = extended_coord.view(nframes, -1).shape[1] // 3
 
-        if not parrallel_mode and self.use_loc_mapping:
+        if not parallel_mode and self.use_loc_mapping:
             node_ebd_ext = self.type_embedding(extended_atype[:, :nloc])
         else:
             node_ebd_ext = self.type_embedding(extended_atype)

deepmd/pt/model/descriptor/repflow_layer.py

@@ -684,7 +684,7 @@ def optim_edge_update_dynamic(
 
     def forward(
         self,
-        node_ebd_ext: torch.Tensor,  # nf x nall x n_dim [OR] nf x nloc x n_dim when not parrallel_mode
+        node_ebd_ext: torch.Tensor,  # nf x nall x n_dim [OR] nf x nloc x n_dim when not parallel_mode
         edge_ebd: torch.Tensor,  # nf x nloc x nnei x e_dim
         h2: torch.Tensor,  # nf x nloc x nnei x 3
         angle_ebd: torch.Tensor,  # nf x nloc x a_nnei x a_nnei x a_dim

deepmd/pt/model/descriptor/repflows.py

@@ -417,8 +417,8 @@ def forward(
         mapping: Optional[torch.Tensor] = None,
         comm_dict: Optional[dict[str, torch.Tensor]] = None,
     ):
-        parrallel_mode = comm_dict is not None
-        if not parrallel_mode:
+        parallel_mode = comm_dict is not None
+        if not parallel_mode:
             assert mapping is not None
         nframes, nloc, nnei = nlist.shape
         nall = extended_coord.view(nframes, -1).shape[1] // 3
@@ -492,7 +492,7 @@ def forward(
         cosine_ij = torch.matmul(normalized_diff_i, normalized_diff_j) * (1 - 1e-6)
         angle_input = cosine_ij.unsqueeze(-1) / (torch.pi**0.5)
 
-        if not parrallel_mode and self.use_loc_mapping:
+        if not parallel_mode and self.use_loc_mapping:
             assert mapping is not None
             # convert nlist from nall to nloc index
             nlist = torch.gather(
@@ -534,15 +534,15 @@ def forward(
         angle_ebd = self.angle_embd(angle_input)
 
         # nb x nall x n_dim
-        if not parrallel_mode:
+        if not parallel_mode:
             assert mapping is not None
             mapping = (
                 mapping.view(nframes, nall).unsqueeze(-1).expand(-1, -1, self.n_dim)
             )
         for idx, ll in enumerate(self.layers):
             # node_ebd:     nb x nloc x n_dim
-            # node_ebd_ext: nb x nall x n_dim [OR] nb x nloc x n_dim when not parrallel_mode
-            if not parrallel_mode:
+            # node_ebd_ext: nb x nall x n_dim [OR] nb x nloc x n_dim when not parallel_mode
+            if not parallel_mode:
                 assert mapping is not None
                 node_ebd_ext = (
                     torch.gather(node_ebd, 1, mapping)