From 234dd375bf93fcda46469d1d87a5e2c39c37bbbc Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Mon, 5 Jun 2023 17:11:48 +0530 Subject: [PATCH 1/2] fix: assertion. --- tests/pipelines/unidiffuser/test_unidiffuser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/unidiffuser/test_unidiffuser.py b/tests/pipelines/unidiffuser/test_unidiffuser.py index 06cb451281c9..346375aa740c 100644 --- a/tests/pipelines/unidiffuser/test_unidiffuser.py +++ b/tests/pipelines/unidiffuser/test_unidiffuser.py @@ -639,7 +639,7 @@ def test_unidiffuser_default_joint_v1_fp16(self): expected_img_slice = np.array([0.2402, 0.2375, 0.2285, 0.2378, 0.2407, 0.2263, 0.2354, 0.2307, 0.2520]) assert np.abs(image_slice.flatten() - expected_img_slice).max() < 2e-1 - expected_text_prefix = "a living room" + expected_text_prefix = "A living room" assert text[0][: len(expected_text_prefix)] == expected_text_prefix def test_unidiffuser_default_text2img_v1_fp16(self): From 69d7f4d61ad9d7e22dbd958eb1b6082b4e78eb4d Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Mon, 5 Jun 2023 17:19:45 +0530 Subject: [PATCH 2/2] assertion fix. --- tests/pipelines/unidiffuser/test_unidiffuser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/unidiffuser/test_unidiffuser.py b/tests/pipelines/unidiffuser/test_unidiffuser.py index 346375aa740c..06cb451281c9 100644 --- a/tests/pipelines/unidiffuser/test_unidiffuser.py +++ b/tests/pipelines/unidiffuser/test_unidiffuser.py @@ -639,7 +639,7 @@ def test_unidiffuser_default_joint_v1_fp16(self): expected_img_slice = np.array([0.2402, 0.2375, 0.2285, 0.2378, 0.2407, 0.2263, 0.2354, 0.2307, 0.2520]) assert np.abs(image_slice.flatten() - expected_img_slice).max() < 2e-1 - expected_text_prefix = "A living room" + expected_text_prefix = "a living room" assert text[0][: len(expected_text_prefix)] == expected_text_prefix def test_unidiffuser_default_text2img_v1_fp16(self):