Fix return_attention_scores bug (#1977)

* Fix `return_attention_scores` bug * Keep the if...else block
keras-team · Nov 11, 2024 · d97db05 · d97db05
1 parent 76a6ca0
commit d97db05
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/keras_hub/src/layers/modeling/transformer_encoder.py b/keras_hub/src/layers/modeling/transformer_encoder.py
@@ -215,7 +215,6 @@ def call(
                 return_attention_scores=return_attention_scores,
                 training=training,
             )
-            return x, attention_scores
         else:
             x = self._self_attention_layer(
                 query=x,