Skip to content

Commit 451790f

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

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
@@ -21,7 +21,7 @@
2121
from diffusers.models.attention import FreeNoiseTransformerBlock
2222
from diffusers.utils import logging
2323
from diffusers.utils.import_utils import is_xformers_available
24-
from diffusers.utils.testing_utils import require_non_cpu, torch_device
24+
from diffusers.utils.testing_utils import require_accelerator, torch_device
2525

2626
from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_PARAMS
2727
from ..test_pipelines_common import (

0 commit comments

Comments
 (0)