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 a138d71 commit 7536f64Copy full SHA for 7536f64
tests/pipelines/kandinsky2_2/test_kandinsky_controlnet.py
@@ -29,6 +29,7 @@
29
)
30
31
from ...testing_utils import (
32
+ Expectations,
33
backend_empty_cache,
34
enable_full_determinism,
35
floats_tensor,
@@ -290,4 +291,11 @@ def test_kandinsky_controlnet(self):
290
291
292
assert image.shape == (512, 512, 3)
293
max_diff = numpy_cosine_similarity_distance(expected_image.flatten(), image.flatten())
- assert max_diff < 2e-4
294
+ expected_max_diffs = Expectations(
295
+ {
296
+ ("xpu", 3): 2e-3,
297
+ ("cuda", 7): 2e-4,
298
+ }
299
+ )
300
+ expected_max_diff = expected_max_diffs.get_expectation()
301
+ assert max_diff < expected_max_diff
0 commit comments