riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Fix CI issues (#35662)

Browse Source 
* make explicit gpu dep

* [run-slow] bamba
This commit is contained in:
Pablo Montalvo 2025-02-11 18:17:01 +01:00 committed by GitHub
parent 775252abd4
commit 3a5c328fd8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/bamba/test_modeling_bamba.py
View File

@ -22,6 +22,7 @@ import pytest
from transformers import AutoTokenizer, BambaConfig, is_torch_available
from transformers.testing_utils import (
require_torch,
require_torch_gpu,
slow,
torch_device,
)
@ -487,6 +488,7 @@ class BambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
@slow
@require_torch
@require_torch_gpu
class BambaModelIntegrationTest(unittest.TestCase):
model = None
tokenizer = None