diff --git a/tests/models/musicgen/test_modeling_musicgen.py b/tests/models/musicgen/test_modeling_musicgen.py index 3d59becc75..dbae06dbf3 100644 --- a/tests/models/musicgen/test_modeling_musicgen.py +++ b/tests/models/musicgen/test_modeling_musicgen.py @@ -506,6 +506,9 @@ class MusicgenTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin, test_pruning = False # training is not supported yet for MusicGen test_headmasking = 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): self.model_tester = MusicgenTester(self)