From 3a5c328fd846574535222dfb4b3eb31fb6c1102f Mon Sep 17 00:00:00 2001
From: Pablo Montalvo <39954772+molbap@users.noreply.github.com>
Date: Tue, 11 Feb 2025 18:17:01 +0100
Subject: [PATCH] Fix CI issues  (#35662)

* make explicit gpu dep

* [run-slow] bamba
---
 tests/models/bamba/test_modeling_bamba.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/models/bamba/test_modeling_bamba.py b/tests/models/bamba/test_modeling_bamba.py
index 68da2fdf02..f7f9481e3e 100644
--- a/tests/models/bamba/test_modeling_bamba.py
+++ b/tests/models/bamba/test_modeling_bamba.py
@@ -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