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/mpt/test_modeling_mpt.py
View File

@@ -18,7 +18,7 @@ import math
import unittest
from transformers import MptConfig, is_torch_available
from transformers.testing_utils import require_torch, require_torch_gpu, slow, torch_device
from transformers.testing_utils import require_bitsandbytes, require_torch, require_torch_gpu, slow, torch_device
from ...generation.test_utils import GenerationTesterMixin
from ...test_configuration_common import ConfigTester
@@ -430,6 +430,7 @@ class MptModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin,
@slow
@require_torch_gpu
@require_bitsandbytes
class MptIntegrationTests(unittest.TestCase):
def test_generation_8k(self):
model_id = "mosaicml/mpt-7b-8k"