We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e0507b5 commit 97f0e9eCopy full SHA for 97f0e9e
tests/pipelines/animatediff/test_animatediff.py
@@ -278,7 +278,7 @@ def test_inference_batch_single_identical(
278
max_diff = np.abs(to_np(output_batch[0][0]) - to_np(output[0][0])).max()
279
assert max_diff < expected_max_diff
280
281
- @require_non_cpu
+ @require_accelerator
282
def test_to_device(self):
283
components = self.get_dummy_components()
284
pipe = self.pipeline_class(**components)
0 commit comments