SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add @require_bitsandbytes to Aria test_batched_generation (#36192)

Browse Source
This commit is contained in:
ivarflakstad
2025-02-14 15:48:47 +01:00
committed by GitHub
parent bcfc9d795e
commit 7ae7e87a09
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/aria/test_modeling_aria.py
View File

@@ -436,6 +436,7 @@ class AriaForConditionalGenerationIntegrationTest(unittest.TestCase):
@slow @slow
@require_torch @require_torch
@require_vision @require_vision
@require_bitsandbytes
def test_batched_generation(self): def test_batched_generation(self):
model = AriaForConditionalGeneration.from_pretrained("rhymes-ai/Aria", load_in_4bit=True) model = AriaForConditionalGeneration.from_pretrained("rhymes-ai/Aria", load_in_4bit=True)