From de70c8426eaf3ac9c54897ace88408450a4c0e76 Mon Sep 17 00:00:00 2001
From: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com>
Date: Tue, 20 May 2025 14:37:55 +0200
Subject: [PATCH] Disable torchscript tests for
 AriaForConditionalGenerationModelTest (#38225)

Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
---
 tests/models/aria/test_modeling_aria.py | 1 +
 1 file changed, 1 insertion(+)

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):