riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-13 17:48:22 +06:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix wav2vec2 params (#27515)

Browse Source 
Fix test
This commit is contained in:
Zach Mueller 2023-11-15 09:24:03 -05:00 committed by GitHub
parent 48ba1e074f
commit a85ea4b19a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/pytorch/test_pytorch_examples.py
View File

@ -550,9 +550,6 @@ class ExamplesTests(TestCasePlus):
--seed 42 --seed 42
""".split() """.split()
if is_torch_fp16_available_on_device(torch_device):
testargs.append("--fp16")
with patch.object(sys, "argv", testargs): with patch.object(sys, "argv", testargs):
run_wav2vec2_pretraining_no_trainer.main() run_wav2vec2_pretraining_no_trainer.main()
model = Wav2Vec2ForPreTraining.from_pretrained(tmp_dir) model = Wav2Vec2ForPreTraining.from_pretrained(tmp_dir)