Skip to content

Commit e0507b5

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

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/animatediff/test_animatediff_controlnet.py

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

284-
@require_non_cpu
284+
@require_accelerator
285285
def test_to_device(self):
286286
components = self.get_dummy_components()
287287
pipe = self.pipeline_class(**components)

0 commit comments

Comments
 (0)