Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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.

Expand Down
Loading