diff --git a/tests/models/aria/test_modeling_aria.py b/tests/models/aria/test_modeling_aria.py index 4a86e8e79f..6adc93bd5a 100644 --- a/tests/models/aria/test_modeling_aria.py +++ b/tests/models/aria/test_modeling_aria.py @@ -180,6 +180,7 @@ class AriaForConditionalGenerationModelTest(ModelTesterMixin, GenerationTesterMi all_model_classes = (AriaModel, AriaForConditionalGeneration) if is_torch_available() else () test_pruning = False test_head_masking = False + test_torchscript = False _is_composite = True def setUp(self):