We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 186eaf8 commit 6ab2e5cCopy full SHA for 6ab2e5c
keras_hub/src/models/smollm3/smollm3_backbone.py
@@ -65,7 +65,6 @@ def __init__(
65
max_position_embeddings,
66
rope_theta,
67
partial_rotary_factor,
68
- num_hidden_layers,
69
**kwargs,
70
):
71
# === Layers ===
@@ -119,7 +118,7 @@ def __init__(
119
118
hidden_states, position_id_input
120
)
121
122
- for decoder_layer in self.decoder_layers[:num_hidden_layers]:
+ for decoder_layer in self.decoder_layers[:num_layers]:
123
hidden_states = decoder_layer(
124
hidden_states,
125
position_embeddings=position_embeddings,
0 commit comments