Clamp PT

tests/test_modeling_common.py

@@ -709,6 +709,8 @@ class ModelTesterMixin:
                     elif tuple_object is None:
                         return
                     else:
+                        tuple_object.clamp_(-1e12, 1e12)
+                        dict_object.clamp_(-1e12, 1e12)
                         self.assertTrue(
                             torch.allclose(tuple_object, dict_object, atol=1e-5),
                             msg=f"Tuple and dict output are not equal. Difference: {torch.max(torch.abs(tuple_object - dict_object))}",

tests/test_modeling_tf_common.py

@@ -536,6 +536,8 @@ class TFModelTesterMixin:
                 elif tuple_object is None:
                     return
                 else:
+                    tuple_object = tf.clip_by_value(tuple_object, -1e12, 1e12)
+                    dict_object = tf.clip_by_value(dict_object, -1e12, 1e12)
                     self.assertTrue(
                         all(tf.equal(tuple_object, dict_object)),
                         msg=f"Tuple and dict output are not equal. Difference: {tf.math.reduce_max(tf.abs(tuple_object - dict_object))}",