Skip to content

Commit d5823a4

Browse files
committed
style: fix EXAONE review cleanup whitespace
Signed-off-by: Bias92 <pewpewplay315@gmail.com>
1 parent 319a6eb commit d5823a4

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

‎src/megatron/bridge/models/gemma/gemma3_provider.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -404,5 +404,3 @@ def _is_local_attn_layer(
404404
) -> bool:
405405
pattern_size = sum(layer_pattern)
406406
return layer_number % pattern_size != 0
407-
408-

‎tests/unit_tests/models/exaone/test_exaone4_bridge.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -155,8 +155,9 @@ def test_mapping_registry_contains_exaone_weights(self):
155155
registry.megatron_to_hf_lookup("decoder.layers.0.self_attention.k_layernorm.weight").hf_param
156156
== "model.layers.0.self_attn.k_norm.weight"
157157
)
158+
post_attn_key = "decoder.layers.0.self_attention.linear_proj.post_layernorm.weight"
158159
assert (
159-
registry.megatron_to_hf_lookup("decoder.layers.0.self_attention.linear_proj.post_layernorm.weight").hf_param
160+
registry.megatron_to_hf_lookup(post_attn_key).hf_param
160161
== "model.layers.0.post_attention_layernorm.weight"
161162
)
162163
assert (

0 commit comments

Comments
 (0)