remove unused statement

caic99 · caic99 · commit 9025d0b313ca · 2025-05-15T09:19:01.000Z
diff --git a/deepmd/pt/model/descriptor/repflow_layer.py b/deepmd/pt/model/descriptor/repflow_layer.py
@@ -682,7 +682,6 @@ def forward(
                     )
                 )
 
-            a_sw.to(edge_angle_update.dtype)
             # nb x nloc x a_nnei x a_nnei x e_dim
             weighted_edge_angle_update = (
                 a_sw[..., None, None] * a_sw[..., None, :, None] * edge_angle_update

Original file line number	Diff line number	Diff line change
`@@ -682,7 +682,6 @@ def forward(`
`682`	`682`	`)`
`683`	`683`	`)`
`684`	`684`
`685`		`- a_sw.to(edge_angle_update.dtype)`
`686`	`685`	`# nb x nloc x a_nnei x a_nnei x e_dim`
`687`	`686`	`weighted_edge_angle_update = (`
`688`	`687`	`a_sw[..., None, None] * a_sw[..., None, :, None] * edge_angle_update`