Skip to content

GitLab

Unverified Commit a85ea4b1 authored by Zach Mueller's avatar Zach Mueller Committed by GitHub
Browse files

Fix wav2vec2 params (#27515) 

Fix test
parent 48ba1e07
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
examples/pytorch/test_pytorch_examples.py
View file @ a85ea4b1
...@@ -550,9 +550,6 @@ class ExamplesTests(TestCasePlus): ...@@ -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)
......
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