Skip to content

Commit 6ab2e5c

Browse files
committed
remove unnecessary arg
1 parent 186eaf8 commit 6ab2e5c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

keras_hub/src/models/smollm3/smollm3_backbone.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,6 @@ def __init__(
6565
max_position_embeddings,
6666
rope_theta,
6767
partial_rotary_factor,
68-
num_hidden_layers,
6968
**kwargs,
7069
):
7170
# === Layers ===
@@ -119,7 +118,7 @@ def __init__(
119118
hidden_states, position_id_input
120119
)
121120

122-
for decoder_layer in self.decoder_layers[:num_hidden_layers]:
121+
for decoder_layer in self.decoder_layers[:num_layers]:
123122
hidden_states = decoder_layer(
124123
hidden_states,
125124
position_embeddings=position_embeddings,

0 commit comments

Comments
 (0)