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

[MPT] Add require_bitsandbytes on MPT integration tests (#25201)

Browse Source 
* add  `require_bitsandbytes` on MPT integration tests

* add it on mpt as well
This commit is contained in:
Younes Belkada
2023-08-01 12:20:34 +02:00
committed by GitHub
parent 972fdcc778
commit 05ebb0264e
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/instructblip/test_modeling_instructblip.py
View File

@@ -29,7 +29,7 @@ from transformers import (
InstructBlipQFormerConfig,
InstructBlipVisionConfig,
)
from transformers.testing_utils import require_torch, require_vision, slow, torch_device
from transformers.testing_utils import require_bitsandbytes, require_torch, require_vision, slow, torch_device
from transformers.utils import is_torch_available, is_vision_available
from ...test_configuration_common import ConfigTester
@@ -521,6 +521,7 @@ def prepare_img():
@require_torch
@slow
class InstructBlipModelIntegrationTest(unittest.TestCase):
@require_bitsandbytes
def test_inference_vicuna_7b(self):
processor = InstructBlipProcessor.from_pretrained("Salesforce/instructblip-vicuna-7b")
model = InstructBlipForConditionalGeneration.from_pretrained(