add cuda dep for a test (#2820)

* add cuda dep for a test * hmmm
huggingface · Jun 3, 2024 · b7fa2fa · b7fa2fa
1 parent d5d378d
commit b7fa2fa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()