Skip to content

Commit bc03727

Browse files
author
Huy Vu2
committed
fix ruff lint
1 parent e332cb2 commit bc03727

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

dfm/src/megatron/model/wan/wan_model.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,6 @@ def __init__(
165165
if self.post_process:
166166
self.head = Head(self.config.hidden_size, self.out_channels, self.patch_size, eps=1e-6)
167167

168-
169168
# set attributes "average_gradients_across_tp_domain" for nn.Parameter objects
170169
# this is used for gradient averaging across TP domain with sequence parallelism
171170
self._mark_trainable_params_for_tp_grad_avg(

0 commit comments

Comments
 (0)