Enable numpy behaviour across the TF codebase

2025-10-21 01:23:56 +08:00 · 2022-09-29 14:45:13 +01:00
1 changed files with 2 additions and 0 deletions
--- a/src/transformers/modeling_tf_utils.py
+++ b/src/transformers/modeling_tf_utils.py
@ -84,6 +84,8 @@ TFModelInputType = Union[
    KerasTensor,
 ]

+tf.experimental.numpy.experimental_enable_numpy_behavior(prefer_float32=True)
+

 def dummy_loss(y_true, y_pred):
    return tf.reduce_mean(y_pred)