From 717dadc6f36be9f50abc66adfd918f9b0e6e3502 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 13 Jul 2023 15:54:18 +0200 Subject: [PATCH] Skip torchscript tests for `MusicgenForConditionalGeneration` (#24782) fix Co-authored-by: ydshieh --- tests/models/musicgen/test_modeling_musicgen.py | 3 +++ 1 file changed, 3 insertions(+) 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)