From 62255b43d451f4d15afec1ea5a9fc9d742702b4c Mon Sep 17 00:00:00 2001 From: Sylvain Gugger Date: Tue, 9 Nov 2021 18:32:29 -0500 Subject: [PATCH] Support for TF >= 2.7 --- setup.py | 4 ++-- src/transformers/dependency_versions_table.py | 4 ++-- src/transformers/models/roformer/modeling_tf_roformer.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/setup.py b/setup.py index 470b82a58855..4ace56bf69f4 100644 --- a/setup.py +++ b/setup.py @@ -151,8 +151,8 @@ "sphinxext-opengraph==0.4.1", "sphinx-intl", "starlette", - "tensorflow-cpu>=2.3,<2.7", - "tensorflow>=2.3,<2.7", + "tensorflow-cpu>=2.3", + "tensorflow>=2.3", "timeout-decorator", "timm", "tokenizers>=0.10.1,<0.11", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 84d0ac09f429..42d9ce1bbe0f 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -62,8 +62,8 @@ "sphinxext-opengraph": "sphinxext-opengraph==0.4.1", "sphinx-intl": "sphinx-intl", "starlette": "starlette", - "tensorflow-cpu": "tensorflow-cpu>=2.3,<2.7", - "tensorflow": "tensorflow>=2.3,<2.7", + "tensorflow-cpu": "tensorflow-cpu>=2.3", + "tensorflow": "tensorflow>=2.3", "timeout-decorator": "timeout-decorator", "timm": "timm", "tokenizers": "tokenizers>=0.10.1,<0.11", diff --git a/src/transformers/models/roformer/modeling_tf_roformer.py b/src/transformers/models/roformer/modeling_tf_roformer.py index ad0a42ff4e7f..915fc02779ad 100644 --- a/src/transformers/models/roformer/modeling_tf_roformer.py +++ b/src/transformers/models/roformer/modeling_tf_roformer.py @@ -1060,7 +1060,7 @@ class TFRoFormerClassificationHead(tf.keras.layers.Layer): """Head for sentence-level classification tasks.""" def __init__(self, config: RoFormerConfig, *inputs, **kwargs): - super().__init__(config, *inputs, **kwargs) + super().__init__(*inputs, **kwargs) self.dense = tf.keras.layers.Dense( units=config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name="dense"