We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent abbf3c1 commit 744663fCopy full SHA for 744663f
tests/pipelines/unclip/test_unclip.py
@@ -358,7 +358,7 @@ class DummyScheduler:
358
def test_attention_slicing_forward_pass(self):
359
test_max_difference = torch_device == "cpu"
360
361
- 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)
362
363
# Overriding PipelineTesterMixin::test_inference_batch_single_identical
364
# because UnCLIP undeterminism requires a looser check.
0 commit comments