Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
7a787c68
Unverified
Commit
7a787c68
authored
Jan 16, 2022
by
Patrick von Platen
Committed by
GitHub
Jan 16, 2022
Browse files
[Speech models] Disable non-existing chunking in tests (#15163)
parent
669e3c50
Changes
6
Show whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
36 additions
and
0 deletions
+36
-0
tests/test_modeling_hubert.py
tests/test_modeling_hubert.py
+8
-0
tests/test_modeling_sew.py
tests/test_modeling_sew.py
+4
-0
tests/test_modeling_sew_d.py
tests/test_modeling_sew_d.py
+4
-0
tests/test_modeling_unispeech.py
tests/test_modeling_unispeech.py
+4
-0
tests/test_modeling_unispeech_sat.py
tests/test_modeling_unispeech_sat.py
+8
-0
tests/test_modeling_wav2vec2.py
tests/test_modeling_wav2vec2.py
+8
-0
No files found.
tests/test_modeling_hubert.py
View file @
7a787c68
...
@@ -425,6 +425,10 @@ class HubertModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -425,6 +425,10 @@ class HubertModelTest(ModelTesterMixin, unittest.TestCase):
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
HubertModel
.
from_pretrained
(
"facebook/hubert-base-ls960"
)
model
=
HubertModel
.
from_pretrained
(
"facebook/hubert-base-ls960"
)
...
@@ -572,6 +576,10 @@ class HubertRobustModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -572,6 +576,10 @@ class HubertRobustModelTest(ModelTesterMixin, unittest.TestCase):
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
HubertModel
.
from_pretrained
(
"facebook/hubert-large-ls960-ft"
)
model
=
HubertModel
.
from_pretrained
(
"facebook/hubert-large-ls960-ft"
)
...
...
tests/test_modeling_sew.py
View file @
7a787c68
...
@@ -428,6 +428,10 @@ class SEWModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -428,6 +428,10 @@ class SEWModelTest(ModelTesterMixin, unittest.TestCase):
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
SEWModel
.
from_pretrained
(
"asapp/sew-tiny-100k"
)
model
=
SEWModel
.
from_pretrained
(
"asapp/sew-tiny-100k"
)
...
...
tests/test_modeling_sew_d.py
View file @
7a787c68
...
@@ -441,6 +441,10 @@ class SEWDModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -441,6 +441,10 @@ class SEWDModelTest(ModelTesterMixin, unittest.TestCase):
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
if
hasattr
(
module
,
"masked_spec_embed"
)
and
module
.
masked_spec_embed
is
not
None
:
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
module
.
masked_spec_embed
.
data
.
fill_
(
3
)
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
SEWDModel
.
from_pretrained
(
"asapp/sew-d-tiny-100k"
)
model
=
SEWDModel
.
from_pretrained
(
"asapp/sew-d-tiny-100k"
)
...
...
tests/test_modeling_unispeech.py
View file @
7a787c68
...
@@ -518,6 +518,10 @@ class UniSpeechRobustModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -518,6 +518,10 @@ class UniSpeechRobustModelTest(ModelTesterMixin, unittest.TestCase):
self
.
assertEqual
(
logits
.
shape
,
(
1
,
1498
,
32
))
self
.
assertEqual
(
logits
.
shape
,
(
1
,
1498
,
32
))
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
UniSpeechModel
.
from_pretrained
(
"microsoft/unispeech-large-1500h-cv"
)
model
=
UniSpeechModel
.
from_pretrained
(
"microsoft/unispeech-large-1500h-cv"
)
...
...
tests/test_modeling_unispeech_sat.py
View file @
7a787c68
...
@@ -541,6 +541,10 @@ class UniSpeechSatModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -541,6 +541,10 @@ class UniSpeechSatModelTest(ModelTesterMixin, unittest.TestCase):
self
.
assertEqual
(
logits
.
shape
,
(
4
,
1498
,
32
))
self
.
assertEqual
(
logits
.
shape
,
(
4
,
1498
,
32
))
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
UniSpeechSatModel
.
from_pretrained
(
"microsoft/unispeech-sat-base-plus"
)
model
=
UniSpeechSatModel
.
from_pretrained
(
"microsoft/unispeech-sat-base-plus"
)
...
@@ -776,6 +780,10 @@ class UniSpeechSatRobustModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -776,6 +780,10 @@ class UniSpeechSatRobustModelTest(ModelTesterMixin, unittest.TestCase):
self
.
assertEqual
(
logits
.
shape
,
(
1
,
1498
,
32
))
self
.
assertEqual
(
logits
.
shape
,
(
1
,
1498
,
32
))
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
UniSpeechSatModel
.
from_pretrained
(
"microsoft/unispeech-sat-large"
)
model
=
UniSpeechSatModel
.
from_pretrained
(
"microsoft/unispeech-sat-large"
)
...
...
tests/test_modeling_wav2vec2.py
View file @
7a787c68
...
@@ -605,6 +605,10 @@ class Wav2Vec2ModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -605,6 +605,10 @@ class Wav2Vec2ModelTest(ModelTesterMixin, unittest.TestCase):
self
.
assertEqual
(
logits
.
shape
,
(
4
,
1498
,
32
))
self
.
assertEqual
(
logits
.
shape
,
(
4
,
1498
,
32
))
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
Wav2Vec2Model
.
from_pretrained
(
"facebook/wav2vec2-base-960h"
)
model
=
Wav2Vec2Model
.
from_pretrained
(
"facebook/wav2vec2-base-960h"
)
...
@@ -901,6 +905,10 @@ class Wav2Vec2RobustModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -901,6 +905,10 @@ class Wav2Vec2RobustModelTest(ModelTesterMixin, unittest.TestCase):
self
.
assertEqual
(
logits
.
shape
,
(
1
,
1498
,
32
))
self
.
assertEqual
(
logits
.
shape
,
(
1
,
1498
,
32
))
@
unittest
.
skip
(
reason
=
"Feed forward chunking is not implemented"
)
def
test_feed_forward_chunking
(
self
):
pass
@
slow
@
slow
def
test_model_from_pretrained
(
self
):
def
test_model_from_pretrained
(
self
):
model
=
Wav2Vec2Model
.
from_pretrained
(
"facebook/wav2vec2-base-960h"
)
model
=
Wav2Vec2Model
.
from_pretrained
(
"facebook/wav2vec2-base-960h"
)
...
...
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