diff --git a/tests/modular_pipelines/test_modular_pipelines_custom_blocks.py b/tests/modular_pipelines/test_modular_pipelines_custom_blocks.py index 3bfedd92b2..20f65740d5 100644 --- a/tests/modular_pipelines/test_modular_pipelines_custom_blocks.py +++ b/tests/modular_pipelines/test_modular_pipelines_custom_blocks.py @@ -36,7 +36,7 @@ from diffusers.modular_pipelines import ( ) from diffusers.utils import logging -from ..testing_utils import CaptureLogger, nightly, require_torch, require_torch_accelerator, slow, torch_device +from ..testing_utils import CaptureLogger, nightly, require_torch_accelerator, require_torch_gpu, slow, torch_device def _create_tiny_model_dir(model_dir): @@ -722,7 +722,7 @@ class TestBlockKwargsTypeInputs: @slow @nightly -@require_torch +@require_torch_gpu class TestKreaCustomBlocksIntegration: repo_id = "krea/krea-realtime-video"