Hotfix - make torchaudio get the correct version in torch_and_flax_job (#28899)

* check

* check

* check

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-02-06 21:00:42 +01:00
committed by GitHub
parent 4830f26965
commit 40658be461
2 changed files with 6 additions and 0 deletions

View File

@@ -533,6 +533,8 @@ class FastSpeech2ConformerWithHifiGanTester:
return config, inputs_dict
@require_torch_accelerator
@require_torch
class FastSpeech2ConformerWithHifiGanTest(ModelTesterMixin, unittest.TestCase):
all_model_classes = (FastSpeech2ConformerWithHifiGan,) if is_torch_available() else ()
test_pruning = False