From a85ea4b19a47824a8e85d3304a698e2d5c8325ec Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Wed, 15 Nov 2023 09:24:03 -0500 Subject: [PATCH] Fix wav2vec2 params (#27515) Fix test --- examples/pytorch/test_pytorch_examples.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/examples/pytorch/test_pytorch_examples.py b/examples/pytorch/test_pytorch_examples.py index 7d27804a7..a0781b356 100644 --- a/examples/pytorch/test_pytorch_examples.py +++ b/examples/pytorch/test_pytorch_examples.py @@ -550,9 +550,6 @@ class ExamplesTests(TestCasePlus): --seed 42 """.split() - if is_torch_fp16_available_on_device(torch_device): - testargs.append("--fp16") - with patch.object(sys, "argv", testargs): run_wav2vec2_pretraining_no_trainer.main() model = Wav2Vec2ForPreTraining.from_pretrained(tmp_dir)