Add @require_bitsandbytes to Aria test_batched_generation (#36192)

2025-02-14 15:48:47 +01:00
parent bcfc9d795e
commit 7ae7e87a09
1 changed files with 1 additions and 0 deletions
--- 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)