remove unused temporal_self_attention attribute from DiTWithAdaLNSubmodules class

sajadn · sajadn · commit 252047a85f73 · 2025-11-07T10:52:05.000-08:00
diff --git a/dfm/src/megatron/model/dit/dit_layer_spec.py b/dfm/src/megatron/model/dit/dit_layer_spec.py
@@ -46,7 +46,6 @@
 
 @dataclass
 class DiTWithAdaLNSubmodules(TransformerLayerSubmodules):
-    temporal_self_attention: Union[ModuleSpec, type] = IdentityOp
     full_self_attention: Union[ModuleSpec, type] = IdentityOp
 
 
