fix #11724 (#11897)

src/transformers/models/marian/modeling_tf_marian.py

@@ -151,11 +151,12 @@ class TFMarianSinusoidalPositionalEmbedding(tf.keras.layers.Layer):
         position_enc = np.array(
             [[pos / np.power(10000, 2 * (j // 2) / dim) for j in range(dim)] for pos in range(n_pos)]
         )
+        table = np.zeros_like(position_enc)
         # index 0 is all zero
-        position_enc[:, 0 : dim // 2] = np.sin(position_enc[:, 0::2])
-        position_enc[:, dim // 2 :] = np.cos(position_enc[:, 1::2])
+        table[:, 0 : dim // 2] = np.sin(position_enc[:, 0::2])
+        table[:, dim // 2 :] = np.cos(position_enc[:, 1::2])
         # convert to tensor
-        table = tf.convert_to_tensor(position_enc)
+        table = tf.convert_to_tensor(table)
         tf.stop_gradient(table)
         return table
 

src/transformers/models/pegasus/modeling_tf_pegasus.py

@@ -152,11 +152,12 @@ class TFPegasusSinusoidalPositionalEmbedding(tf.keras.layers.Layer):
         position_enc = np.array(
             [[pos / np.power(10000, 2 * (j // 2) / dim) for j in range(dim)] for pos in range(n_pos)]
         )
+        table = np.zeros_like(position_enc)
         # index 0 is all zero
-        position_enc[:, 0 : dim // 2] = np.sin(position_enc[:, 0::2])
-        position_enc[:, dim // 2 :] = np.cos(position_enc[:, 1::2])
+        table[:, 0 : dim // 2] = np.sin(position_enc[:, 0::2])
+        table[:, dim // 2 :] = np.cos(position_enc[:, 1::2])
         # convert to tensor
-        table = tf.convert_to_tensor(position_enc)
+        table = tf.convert_to_tensor(table)
         tf.stop_gradient(table)
         return table