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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user