diff --git a/tests/pipelines/dance_diffusion/test_dance_diffusion.py b/tests/pipelines/dance_diffusion/test_dance_diffusion.py index a63ef84c63..c342e26b77 100644 --- a/tests/pipelines/dance_diffusion/test_dance_diffusion.py +++ b/tests/pipelines/dance_diffusion/test_dance_diffusion.py @@ -73,7 +73,7 @@ class PipelineFastTests(unittest.TestCase): audio_from_tuple_slice = audio_from_tuple[0, -3:, -3:] assert audio.shape == (1, 2, self.dummy_unet.sample_size) - expected_slice = np.array([-0.7265, 1.0000, -0.8388, 0.1175, 0.9498, -1.0000]) + expected_slice = np.array([-0.8682, 1.0000, -1.0000, -0.3130, 0.5133, -1.0000]) assert np.abs(audio_slice.flatten() - expected_slice).max() < 1e-2 assert np.abs(audio_from_tuple_slice.flatten() - expected_slice).max() < 1e-2