fix fast test (#3241)

744663f8 · Patrick von Platen · GitHub · abbf3c1a · 744663f8
Unverified Commit 744663f8 authored Apr 26, 2023 by Patrick von Platen Committed by GitHub Apr 26, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/pipelines/unclip/test_unclip.py tests/pipelines/unclip/test_unclip.py +1 -1

No files found.
--- a/tests/pipelines/unclip/test_unclip.py
+++ b/tests/pipelines/unclip/test_unclip.py
@@ -358,7 +358,7 @@ class UnCLIPPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
    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.