Skip to content

Commit d0e2ab3

Browse files
Apply suggestions from code review
1 parent f0f12d6 commit d0e2ab3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/openvino/test_diffusion.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -140,11 +140,10 @@ def test_compare_to_diffusers_pipeline(self, model_arch: str):
140140
if model_arch == "sana":
141141
# resolution binning will lead to resize output to standard resolution and back that can interpolate floating-point deviations
142142
inputs["use_resolution_binning"] = False
143-
atol = 1e-4
144143

145144
ov_output = ov_pipeline(**inputs, generator=get_generator("pt", SEED)).images
146145
diffusers_output = diffusers_pipeline(**inputs, generator=get_generator("pt", SEED)).images
147-
np.testing.assert_allclose(ov_output, diffusers_output, atol=atol, rtol=1e-2)
146+
np.testing.assert_allclose(ov_output, diffusers_output, atol=6e-3, rtol=1e-2)
148147

149148
# test on inputs nondivisible on 64
150149
height, width, batch_size = 96, 96, 1

0 commit comments

Comments
 (0)