mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-27 17:22:53 +03:00
fix fast test (#3241)
This commit is contained in:
committed by
GitHub
parent
abbf3c1adf
commit
744663f8dc
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user