SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Support for TF >= 2.7 (#14345)

Browse Source
This commit is contained in:
Sylvain Gugger
2021-11-09 18:49:29 -05:00
committed by GitHub
parent e81d8d7fa9
commit 46d0cdae40
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@@ -151,8 +151,8 @@ _deps = [
"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",

4
src/transformers/dependency_versions_table.py
View File

@@ -62,8 +62,8 @@ deps = {
"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",

2
src/transformers/models/roformer/modeling_tf_roformer.py
View File

@@ -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"