add more

faaany · faaany · commit 961a8e5b10b4 · 2024-09-13T00:34:19.000-07:00
diff --git a/tests/models/test_modeling_common.py b/tests/models/test_modeling_common.py
@@ -753,7 +753,7 @@ def test_deprecated_kwargs(self):
                 " from `_deprecated_kwargs = [<deprecated_argument>]`"
             )
 
-    @require_torch_gpu
+    @require_torch_accelerator
     def test_cpu_offload(self):
         config, inputs_dict = self.prepare_init_args_and_inputs_for_common()
         model = self.model_class(**config).eval()

Original file line number	Diff line number	Diff line change
`@@ -753,7 +753,7 @@ def test_deprecated_kwargs(self):`
`753`	`753`	" from `_deprecated_kwargs = [<deprecated_argument>]`"
`754`	`754`	`)`
`755`	`755`
`756`		`- @require_torch_gpu`
	`756`	`+ @require_torch_accelerator`
`757`	`757`	`def test_cpu_offload(self):`
`758`	`758`	`config, inputs_dict = self.prepare_init_args_and_inputs_for_common()`
`759`	`759`	`model = self.model_class(**config).eval()`