From b7fa2fa956f40e0b6f650d5eb1764680bf3fd8f7 Mon Sep 17 00:00:00 2001
From: Marc Sun <57196510+SunMarc@users.noreply.github.com>
Date: Mon, 3 Jun 2024 14:37:44 +0200
Subject: [PATCH] add cuda dep for a test (#2820)

* add cuda dep for a test

* hmmm
---
 tests/test_accelerator.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_accelerator.py b/tests/test_accelerator.py
index aaee1194753..df11f5488d8 100644
--- a/tests/test_accelerator.py
+++ b/tests/test_accelerator.py
@@ -282,6 +282,7 @@ def test_save_model_offload(self, use_safetensors):
         assert torch.allclose(expected, output, atol=1e-5)
 
     @parameterized.expand([True, False], name_func=parameterized_custom_name_func)
+    @require_cuda
     def test_get_state_dict_from_offload(self, use_safetensors):
         accelerator = Accelerator()