From 0dff5869643f0a4d7d7155218fe6bd02e7212208 Mon Sep 17 00:00:00 2001 From: Nathan Lambert Date: Tue, 29 Nov 2022 15:21:38 -0800 Subject: [PATCH] revert dance diff test --- tests/pipelines/dance_diffusion/test_dance_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/dance_diffusion/test_dance_diffusion.py b/tests/pipelines/dance_diffusion/test_dance_diffusion.py index c342e26b77..f1b58fe1c2 100644 --- a/tests/pipelines/dance_diffusion/test_dance_diffusion.py +++ b/tests/pipelines/dance_diffusion/test_dance_diffusion.py @@ -118,5 +118,5 @@ class PipelineIntegrationTests(unittest.TestCase): audio_slice = audio[0, -3:, -3:] assert audio.shape == (1, 2, pipe.unet.sample_size) - expected_slice = np.array([-0.1693, -0.1698, -0.1447, -0.3044, -0.3203, -0.2937]) + expected_slice = np.array([-0.7265, 1.0000, -0.8388, 0.1175, 0.9498, -1.0000]) assert np.abs(audio_slice.flatten() - expected_slice).max() < 1e-2