From 7ae7e87a09b23e83adc9492d69e2546a755c7b59 Mon Sep 17 00:00:00 2001
From: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com>
Date: Fri, 14 Feb 2025 15:48:47 +0100
Subject: [PATCH] Add @require_bitsandbytes to Aria test_batched_generation
 (#36192)

---
 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 8b5e62de14..fb74e914bd 100644
--- a/tests/models/aria/test_modeling_aria.py
+++ b/tests/models/aria/test_modeling_aria.py
@@ -436,6 +436,7 @@ class AriaForConditionalGenerationIntegrationTest(unittest.TestCase):
     @slow
     @require_torch
     @require_vision
+    @require_bitsandbytes
     def test_batched_generation(self):
         model = AriaForConditionalGeneration.from_pretrained("rhymes-ai/Aria", load_in_4bit=True)