Romve print()

keras-team · Apr 26, 2024 · 60e0471 · 60e0471
1 parent 4721f3f
commit 60e0471
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/keras_nlp/models/phi3/phi3_rotary_embedding.py b/keras_nlp/models/phi3/phi3_rotary_embedding.py
@@ -105,8 +105,6 @@ def _compute_cos_sin_embedding(self, inputs, start_index=0, positions=None):
         inverse_freq = self._get_inverse_freq(rotary_dim)
 
         # Multiply inverse_freq by a factor.
-        print("before", ops.shape(inverse_freq))
-
         if (
             ops.shape(inputs)[sequence_axis]
             > self.original_max_position_embeddings
@@ -119,8 +117,6 @@ def _compute_cos_sin_embedding(self, inputs, start_index=0, positions=None):
                 inverse_freq, self.inverese_freq_short_factor
             )
 
-        print("after", ops.shape(inverse_freq))
-
         if positions is None:
             positions = self._compute_positions(inputs, start_index)
         else: