Skip to content

Commit 4b6d2a2

Browse files
committed
Fixed final_norm global import
1 parent ae82959 commit 4b6d2a2

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

keras_hub/src/models/modernbert/modernbert_backbone.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -104,12 +104,12 @@ def __init__(
104104
name=f"transformer_layer_{i}",
105105
)
106106
)
107-
self.final_norm = layers.LayerNormalization(
108-
epsilon=layer_norm_epsilon,
109-
rms_scaling=True,
110-
dtype=dtype,
111-
name="final_norm",
112-
)
107+
self.final_norm = layers.LayerNormalization(
108+
epsilon=layer_norm_epsilon,
109+
rms_scaling=True,
110+
dtype=dtype,
111+
name="final_norm",
112+
)
113113

114114
# Functional API Call
115115
token_id_input = keras.Input(shape=(None,), dtype="int32", name="token_ids")

0 commit comments

Comments
 (0)