Skip to content

Commit 97f0e9e

Browse files
faaanyDN6
andauthored
Update tests/pipelines/animatediff/test_animatediff.py
Co-authored-by: Dhruv Nair <[email protected]>
1 parent e0507b5 commit 97f0e9e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/animatediff/test_animatediff.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -278,7 +278,7 @@ def test_inference_batch_single_identical(
278278
max_diff = np.abs(to_np(output_batch[0][0]) - to_np(output[0][0])).max()
279279
assert max_diff < expected_max_diff
280280

281-
@require_non_cpu
281+
@require_accelerator
282282
def test_to_device(self):
283283
components = self.get_dummy_components()
284284
pipe = self.pipeline_class(**components)

0 commit comments

Comments
 (0)