From 731dd5f1d13b11696edd5e1764ac5b4eabd336fd Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Thu, 24 Aug 2023 09:31:00 +0000 Subject: [PATCH] use max to test difference --- .../versatile_diffusion/test_versatile_diffusion_dual_guided.py | 2 +- .../versatile_diffusion/test_versatile_diffusion_mega.py | 2 +- .../test_versatile_diffusion_text_to_image.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py index 4e2b89982a..bb8584192f 100644 --- a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py +++ b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py @@ -76,7 +76,7 @@ class VersatileDiffusionDualGuidedPipelineIntegrationTests(unittest.TestCase): output_type="numpy", ).images - assert np.abs(image - new_image).sum() < 1e-5, "Models don't have the same forward pass" + assert np.abs(image - new_image).max() < 1e-5, "Models don't have the same forward pass" def test_inference_dual_guided(self): pipe = VersatileDiffusionDualGuidedPipeline.from_pretrained("shi-labs/versatile-diffusion") diff --git a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py index b77c1baf41..585f4f023b 100644 --- a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py +++ b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py @@ -77,7 +77,7 @@ class VersatileDiffusionMegaPipelineIntegrationTests(unittest.TestCase): output_type="numpy", ).images - assert np.abs(image - new_image).sum() < 1e-5, "Models don't have the same forward pass" + assert np.abs(image - new_image).max() < 1e-5, "Models don't have the same forward pass" def test_inference_dual_guided_then_text_to_image(self): pipe = VersatileDiffusionPipeline.from_pretrained("shi-labs/versatile-diffusion", torch_dtype=torch.float16) diff --git a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_text_to_image.py b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_text_to_image.py index 194f660f70..e177707784 100644 --- a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_text_to_image.py +++ b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_text_to_image.py @@ -64,7 +64,7 @@ class VersatileDiffusionTextToImagePipelineIntegrationTests(unittest.TestCase): prompt=prompt, generator=generator, guidance_scale=7.5, num_inference_steps=2, output_type="numpy" ).images - assert np.abs(image - new_image).sum() < 1e-5, "Models don't have the same forward pass" + assert np.abs(image - new_image).max() < 1e-5, "Models don't have the same forward pass" def test_inference_text2img(self): pipe = VersatileDiffusionTextToImagePipeline.from_pretrained(