diff --git a/tests/pipelines/unclip/test_unclip.py b/tests/pipelines/unclip/test_unclip.py index d2c699ea501d..5c9181c08e3f 100644 --- a/tests/pipelines/unclip/test_unclip.py +++ b/tests/pipelines/unclip/test_unclip.py @@ -358,7 +358,7 @@ class DummyScheduler: def test_attention_slicing_forward_pass(self): test_max_difference = torch_device == "cpu" - self._test_attention_slicing_forward_pass(test_max_difference=test_max_difference) + self._test_attention_slicing_forward_pass(test_max_difference=test_max_difference, expected_max_diff=0.01) # Overriding PipelineTesterMixin::test_inference_batch_single_identical # because UnCLIP undeterminism requires a looser check.