Unverified Commit 46d0cdae authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Support for TF >= 2.7 (#14345)

parent e81d8d7f
...@@ -151,8 +151,8 @@ _deps = [ ...@@ -151,8 +151,8 @@ _deps = [
"sphinxext-opengraph==0.4.1", "sphinxext-opengraph==0.4.1",
"sphinx-intl", "sphinx-intl",
"starlette", "starlette",
"tensorflow-cpu>=2.3,<2.7", "tensorflow-cpu>=2.3",
"tensorflow>=2.3,<2.7", "tensorflow>=2.3",
"timeout-decorator", "timeout-decorator",
"timm", "timm",
"tokenizers>=0.10.1,<0.11", "tokenizers>=0.10.1,<0.11",
......
...@@ -62,8 +62,8 @@ deps = { ...@@ -62,8 +62,8 @@ deps = {
"sphinxext-opengraph": "sphinxext-opengraph==0.4.1", "sphinxext-opengraph": "sphinxext-opengraph==0.4.1",
"sphinx-intl": "sphinx-intl", "sphinx-intl": "sphinx-intl",
"starlette": "starlette", "starlette": "starlette",
"tensorflow-cpu": "tensorflow-cpu>=2.3,<2.7", "tensorflow-cpu": "tensorflow-cpu>=2.3",
"tensorflow": "tensorflow>=2.3,<2.7", "tensorflow": "tensorflow>=2.3",
"timeout-decorator": "timeout-decorator", "timeout-decorator": "timeout-decorator",
"timm": "timm", "timm": "timm",
"tokenizers": "tokenizers>=0.10.1,<0.11", "tokenizers": "tokenizers>=0.10.1,<0.11",
......
...@@ -1060,7 +1060,7 @@ class TFRoFormerClassificationHead(tf.keras.layers.Layer): ...@@ -1060,7 +1060,7 @@ class TFRoFormerClassificationHead(tf.keras.layers.Layer):
"""Head for sentence-level classification tasks.""" """Head for sentence-level classification tasks."""
def __init__(self, config: RoFormerConfig, *inputs, **kwargs): def __init__(self, config: RoFormerConfig, *inputs, **kwargs):
super().__init__(config, *inputs, **kwargs) super().__init__(*inputs, **kwargs)
self.dense = tf.keras.layers.Dense( self.dense = tf.keras.layers.Dense(
units=config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name="dense" units=config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name="dense"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment