SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
1 changed files with 2 additions and 0 deletions
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 import AutoTokenizer, BambaConfig, is_torch_available
from transformers.testing_utils import ( from transformers.testing_utils import (
require_torch, require_torch,
require_torch_gpu,
slow, slow,
torch_device, torch_device,
) )
@@ -487,6 +488,7 @@ class BambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
@slow @slow
@require_torch @require_torch
@require_torch_gpu
class BambaModelIntegrationTest(unittest.TestCase): class BambaModelIntegrationTest(unittest.TestCase):
model = None model = None
tokenizer = None tokenizer = None