File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
keras_hub/src/models/smollm3 Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -94,6 +94,7 @@ def build(self, input_shape):
94
94
self .k_proj .build (hidden_states_shape )
95
95
self .v_proj .build (hidden_states_shape )
96
96
self .o_proj .build (hidden_states_shape )
97
+ self .training = False
97
98
super ().build (input_shape )
98
99
99
100
def call (
@@ -276,6 +277,7 @@ def build(self, input_shape):
276
277
self .intermediate_size ,
277
278
)
278
279
self .down_proj .build (down_proj_input_shape )
280
+ self .training = False
279
281
super ().build (input_shape )
280
282
281
283
def call (self , x ):
@@ -397,6 +399,7 @@ def build(self, input_shape):
397
399
self .mlp .build (input_shape )
398
400
self .input_layernorm .build (input_shape )
399
401
self .post_attention_layernorm .build (input_shape )
402
+ self .training = False
400
403
401
404
super ().build (input_shape )
402
405
@@ -515,6 +518,7 @@ def build(self, input_shape):
515
518
- position_ids_shape: (batch_size, seq_len)
516
519
"""
517
520
# No internal layers to explicitly build here, as inv_freq is added in __init__
521
+ self .training = False
518
522
super ().build (input_shape )
519
523
520
524
def call (
You can’t perform that action at this time.
0 commit comments