update comment

faaany · faaany · commit 6296e8821b93 · 2024-09-09T18:04:21.000-07:00
diff --git a/tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py b/tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py
@@ -251,7 +251,7 @@ def test_float16_inference(self):
 
     @unittest.skipIf(
         not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),
-        reason="CPU offload is only available with CUDA and `accelerate v0.14.0` or higher",
+        reason="CPU offload is only available with `accelerate v0.14.0` or higher",
     )
     def test_cpu_offload_forward_pass(self):
         components = self.get_dummy_components()

Original file line number	Diff line number	Diff line change
`@@ -251,7 +251,7 @@ def test_float16_inference(self):`
`251`	`251`
`252`	`252`	`@unittest.skipIf(`
`253`	`253`	`not is_accelerate_available() or is_accelerate_version("<", "0.14.0"),`
`254`		- reason="CPU offload is only available with CUDA and `accelerate v0.14.0` or higher",
	`254`	+ reason="CPU offload is only available with `accelerate v0.14.0` or higher",
`255`	`255`	`)`
`256`	`256`	`def test_cpu_offload_forward_pass(self):`
`257`	`257`	`components = self.get_dummy_components()`