Fix references for e2e tests

2024-09-19 19:05:28 +00:00 · 2024-01-19 14:52:05 +01:00 · 2024-01-19 14:52:05 +01:00 · 42b7749630
parent de6266010d
commit 42b7749630
8 changed files with 2 additions and 1 deletions
--- a/tests/e2e/test_diffusion.py
+++ b/tests/e2e/test_diffusion.py
@ -1141,7 +1141,8 @@ def test_diffusion_refonly(
        torch.randn(2, 4, 64, 64, device=test_device)  # for SD Web UI reproductibility only
    predicted_image = sd15.lda.decode_latents(x)

-    ensure_similar_images(predicted_image, expected_image_refonly, min_psnr=35, min_ssim=0.99)
+    # min_psnr lowered to 33 because this reference image was generated without noise removal (see #192)
+    ensure_similar_images(predicted_image, expected_image_refonly, min_psnr=33, min_ssim=0.99)


@no_grad()
--- a/tests/e2e/test_diffusion_ref/expected_cutecat_sdxl_ddim_random_init.png
+++ b/tests/e2e/test_diffusion_ref/expected_cutecat_sdxl_ddim_random_init.png
--- a/tests/e2e/test_diffusion_ref/expected_cutecat_sdxl_ddim_random_init_sag.png
+++ b/tests/e2e/test_diffusion_ref/expected_cutecat_sdxl_ddim_random_init_sag.png
--- a/tests/e2e/test_diffusion_ref/expected_dropy_slime_9752.png
+++ b/tests/e2e/test_diffusion_ref/expected_dropy_slime_9752.png
--- a/tests/e2e/test_diffusion_ref/expected_image_sdxl_ip_adapter_plus_woman.png
+++ b/tests/e2e/test_diffusion_ref/expected_image_sdxl_ip_adapter_plus_woman.png
--- a/tests/e2e/test_diffusion_ref/expected_image_sdxl_ip_adapter_woman.png
+++ b/tests/e2e/test_diffusion_ref/expected_image_sdxl_ip_adapter_woman.png
--- a/tests/e2e/test_diffusion_ref/expected_sdxl_dpo_lora.png
+++ b/tests/e2e/test_diffusion_ref/expected_sdxl_dpo_lora.png
--- a/tests/e2e/test_diffusion_ref/expected_t2i_adapter_xl_canny.png
+++ b/tests/e2e/test_diffusion_ref/expected_t2i_adapter_xl_canny.png