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
ce01122a
Unverified
Commit
ce01122a
authored
Nov 02, 2021
by
Anton Lozhkov
Committed by
GitHub
Nov 02, 2021
Browse files
[Tests] Fix DistilHubert path (#14245)
* Add audio-classification benchmarking results * fix distilhubert path
parent
4a394cf5
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
tests/test_modeling_hubert.py
tests/test_modeling_hubert.py
+2
-2
No files found.
tests/test_modeling_hubert.py
View file @
ce01122a
...
...
@@ -762,8 +762,8 @@ class HubertModelIntegrationTest(unittest.TestCase):
self
.
assertTrue
(
torch
.
allclose
(
predicted_logits
,
expected_logits
,
atol
=
1e-1
))
def
test_inference_distilhubert
(
self
):
model
=
HubertModel
.
from_pretrained
(
"
a
nt
on-
l/distilhubert"
).
to
(
torch_device
)
processor
=
Wav2Vec2FeatureExtractor
.
from_pretrained
(
"
a
nt
on-
l/distilhubert"
)
model
=
HubertModel
.
from_pretrained
(
"nt
u-spm
l/distilhubert"
).
to
(
torch_device
)
processor
=
Wav2Vec2FeatureExtractor
.
from_pretrained
(
"nt
u-spm
l/distilhubert"
)
# TODO: can't test on batched inputs due to incompatible padding https://github.com/pytorch/fairseq/pull/3572
input_speech
=
self
.
_load_datasamples
(
1
)
...
...
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