Unverified Commit 717dadc6 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Skip torchscript tests for `MusicgenForConditionalGeneration` (#24782)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent e367a977
...@@ -506,6 +506,9 @@ class MusicgenTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin, ...@@ -506,6 +506,9 @@ class MusicgenTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin,
test_pruning = False # training is not supported yet for MusicGen test_pruning = False # training is not supported yet for MusicGen
test_headmasking = False test_headmasking = False
test_resize_embeddings = False test_resize_embeddings = False
# not to test torchscript as the model tester doesn't prepare `input_values` and `padding_mask`
# (and `torchscript` hates `None` values).
test_torchscript = False
def setUp(self): def setUp(self):
self.model_tester = MusicgenTester(self) self.model_tester = MusicgenTester(self)
......
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