From ce01122a3bb690a944054bef216f87b90c2c9e63 Mon Sep 17 00:00:00 2001 From: Anton Lozhkov Date: Tue, 2 Nov 2021 17:53:50 +0300 Subject: [PATCH] [Tests] Fix DistilHubert path (#14245) * Add audio-classification benchmarking results * fix distilhubert path --- tests/test_modeling_hubert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_modeling_hubert.py b/tests/test_modeling_hubert.py index 6d69f6c923..61a1b9dfae 100644 --- a/tests/test_modeling_hubert.py +++ b/tests/test_modeling_hubert.py @@ -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("anton-l/distilhubert").to(torch_device) - processor = Wav2Vec2FeatureExtractor.from_pretrained("anton-l/distilhubert") + model = HubertModel.from_pretrained("ntu-spml/distilhubert").to(torch_device) + 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 input_speech = self._load_datasamples(1)