Skip to content

GitLab

Unverified Commit ce01122a authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

[Tests] Fix DistilHubert path (#14245) 

* Add audio-classification benchmarking results

* fix distilhubert path
parent 4a394cf5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_modeling_hubert.py
View file @ ce01122a
...@@ -762,8 +762,8 @@ class HubertModelIntegrationTest(unittest.TestCase): ...@@ -762,8 +762,8 @@ class HubertModelIntegrationTest(unittest.TestCase):
self.assertTrue(torch.allclose(predicted_logits, expected_logits, atol=1e-1)) self.assertTrue(torch.allclose(predicted_logits, expected_logits, atol=1e-1))
def test_inference_distilhubert(self): def test_inference_distilhubert(self):
model = HubertModel.from_pretrained("anton-l/distilhubert").to(torch_device) model = HubertModel.from_pretrained("ntu-spml/distilhubert").to(torch_device)
processor = Wav2Vec2FeatureExtractor.from_pretrained("anton-l/distilhubert") processor = Wav2Vec2FeatureExtractor.from_pretrained("ntu-spml/distilhubert")
# TODO: can't test on batched inputs due to incompatible padding https://github.com/pytorch/fairseq/pull/3572 # TODO: can't test on batched inputs due to incompatible padding https://github.com/pytorch/fairseq/pull/3572
input_speech = self._load_datasamples(1) input_speech = self._load_datasamples(1)
......
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