riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

[Tests] Fix DistilHubert path (#14245)

Browse Source 
* Add audio-classification benchmarking results

* fix distilhubert path
This commit is contained in:
Anton Lozhkov 2021-11-02 17:53:50 +03:00 committed by GitHub
parent 4a394cf53f
commit ce01122a3b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_modeling_hubert.py
View File

@ -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)