"doc/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "e4ab0ebe86d679586b9cde01d0648d3bf5d65860"
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
......@@ -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)
......
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