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
9e72eb44
Commit
9e72eb44
authored
Jun 07, 2022
by
Sylvain Gugger
Browse files
Skip disk offload test for T5
parent
b1187307
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
tests/models/t5/test_modeling_t5.py
tests/models/t5/test_modeling_t5.py
+4
-0
No files found.
tests/models/t5/test_modeling_t5.py
View file @
9e72eb44
...
@@ -662,6 +662,10 @@ class T5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
...
@@ -662,6 +662,10 @@ class T5ModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
attn_weights
=
out
[
attn_name
]
if
attn_name
==
attention_names
[
0
]
else
out
[
attn_name
][
-
1
]
attn_weights
=
out
[
attn_name
]
if
attn_name
==
attention_names
[
0
]
else
out
[
attn_name
][
-
1
]
self
.
assertEqual
(
sum
([
w
.
sum
().
item
()
for
w
in
attn_weights
]),
0.0
)
self
.
assertEqual
(
sum
([
w
.
sum
().
item
()
for
w
in
attn_weights
]),
0.0
)
@
unittest
.
skip
(
"Does not work on the tiny model as we keep hitting edge cases."
)
def
test_disk_offload
(
self
):
pass
class
T5EncoderOnlyModelTester
:
class
T5EncoderOnlyModelTester
:
def
__init__
(
def
__init__
(
...
...
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