From 78a2251f52115317c239f70a10c2143c11fb3abd Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Wed, 12 Aug 2026 21:00:10 -0700 Subject: [PATCH] Add require_accelerator decorator to test_tiles_covering_the_video_match_untiled_exactly to prevent CI timeout on CPU --- .../models/autoencoders/test_models_ltx2_diffusion_decoder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/autoencoders/test_models_ltx2_diffusion_decoder.py b/tests/models/autoencoders/test_models_ltx2_diffusion_decoder.py index e0ae03aa6807..c22eac149c8a 100644 --- a/tests/models/autoencoders/test_models_ltx2_diffusion_decoder.py +++ b/tests/models/autoencoders/test_models_ltx2_diffusion_decoder.py @@ -22,7 +22,7 @@ from diffusers.utils import is_kernels_available from diffusers.utils.torch_utils import randn_tensor -from ...testing_utils import enable_full_determinism, require_torch_gpu, torch_device +from ...testing_utils import enable_full_determinism, require_accelerator, require_torch_gpu, torch_device from ..testing_utils import ( AttentionTesterMixin, BaseModelTesterConfig, @@ -145,6 +145,7 @@ def decode(self, model, latent, num_inference_steps=None): with torch.no_grad(): return model.decode(latent, generator=generator, num_inference_steps=num_inference_steps)[0] + @require_accelerator def test_tiles_covering_the_video_match_untiled_exactly(self): """A tile schedule with a single covering tile must reproduce the untiled decode bit for bit.