Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
46d0cdae
"docs/source/vscode:/vscode.git/clone" did not exist on "eab0afc19ceaf9a31190777f5548312d2346cd44"
Unverified
Commit
46d0cdae
authored
Nov 09, 2021
by
Sylvain Gugger
Committed by
GitHub
Nov 09, 2021
Browse files
Support for TF >= 2.7 (#14345)
parent
e81d8d7f
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
5 deletions
+5
-5
setup.py
setup.py
+2
-2
src/transformers/dependency_versions_table.py
src/transformers/dependency_versions_table.py
+2
-2
src/transformers/models/roformer/modeling_tf_roformer.py
src/transformers/models/roformer/modeling_tf_roformer.py
+1
-1
No files found.
setup.py
View file @
46d0cdae
...
...
@@ -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"
,
...
...
src/transformers/dependency_versions_table.py
View file @
46d0cdae
...
...
@@ -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"
,
...
...
src/transformers/models/roformer/modeling_tf_roformer.py
View file @
46d0cdae
...
...
@@ -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"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment