From 2e618cb6027037597cf80905fc3e356a71923087 Mon Sep 17 00:00:00 2001 From: "Wang, Yi" Date: Mon, 31 Aug 2026 14:08:02 +0800 Subject: [PATCH] =?UTF-8?q?skip=20these=20case=20for=20no-cuda=20device=20?= =?UTF-8?q?since=20the=20modeling=20contains=20cuda=20s=E2=80=A6=20(#14656?= =?UTF-8?q?)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit skip these case for no-cuda device since the modeling contains cuda specific code see https://huggingface.co/krea/krea-realtime-video/blob/main/transformer/model.py#L33 Signed-off-by: Wang, Yi A Co-authored-by: Sayak Paul --- .../modular_pipelines/test_modular_pipelines_custom_blocks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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"