diff --git a/tests/expected_output/test_controlnet[canny]_.png b/tests/expected_output/test_controlnet[canny]_.png index 7aa291a..ceae288 100644 Binary files a/tests/expected_output/test_controlnet[canny]_.png and b/tests/expected_output/test_controlnet[canny]_.png differ diff --git a/tests/expected_output/test_controlnet[depth]_.png b/tests/expected_output/test_controlnet[depth]_.png index d4de258..292d7ca 100644 Binary files a/tests/expected_output/test_controlnet[depth]_.png and b/tests/expected_output/test_controlnet[depth]_.png differ diff --git a/tests/expected_output/test_controlnet[details]_.png b/tests/expected_output/test_controlnet[details]_.png new file mode 100644 index 0000000..0ec5823 Binary files /dev/null and b/tests/expected_output/test_controlnet[details]_.png differ diff --git a/tests/expected_output/test_controlnet[edit]_.png b/tests/expected_output/test_controlnet[edit]_.png new file mode 100644 index 0000000..61fe136 Binary files /dev/null and b/tests/expected_output/test_controlnet[edit]_.png differ diff --git a/tests/expected_output/test_controlnet[hed]_.png b/tests/expected_output/test_controlnet[hed]_.png index 50c122a..d27200a 100644 Binary files a/tests/expected_output/test_controlnet[hed]_.png and b/tests/expected_output/test_controlnet[hed]_.png differ diff --git a/tests/expected_output/test_controlnet[inpaint]_.png b/tests/expected_output/test_controlnet[inpaint]_.png new file mode 100644 index 0000000..3f937e9 Binary files /dev/null and b/tests/expected_output/test_controlnet[inpaint]_.png differ diff --git a/tests/expected_output/test_controlnet[normal]_.png b/tests/expected_output/test_controlnet[normal]_.png index b475fae..b36fec5 100644 Binary files a/tests/expected_output/test_controlnet[normal]_.png and b/tests/expected_output/test_controlnet[normal]_.png differ diff --git a/tests/expected_output/test_controlnet[openpose]_.png b/tests/expected_output/test_controlnet[openpose]_.png index 230cb37..38bcd87 100644 Binary files a/tests/expected_output/test_controlnet[openpose]_.png and b/tests/expected_output/test_controlnet[openpose]_.png differ diff --git a/tests/expected_output/test_controlnet[shuffle]_.png b/tests/expected_output/test_controlnet[shuffle]_.png new file mode 100644 index 0000000..0ec3bb7 Binary files /dev/null and b/tests/expected_output/test_controlnet[shuffle]_.png differ diff --git a/tests/test_clip_embedder.py b/tests/test_clip_embedder.py index 2507a62..08e93d3 100644 --- a/tests/test_clip_embedder.py +++ b/tests/test_clip_embedder.py @@ -26,4 +26,4 @@ def test_text_conditioning(): neutral_embedding_mps_expected = f.get_tensor("neutral_clip_embedding_mps") diff = neutral_embedding - neutral_embedding_mps_expected - assert diff.sum() < 0.05 + assert diff.sum() < 0.09 diff --git a/tests/test_enhancers.py b/tests/test_enhancers.py index 0e30748..45ea0fd 100644 --- a/tests/test_enhancers.py +++ b/tests/test_enhancers.py @@ -134,11 +134,13 @@ def test_clip_mask_parser(mask_text, expected): @pytest.mark.skipif(get_device() == "cpu", reason="Too slow to run on CPU") def test_describe_picture(): + seed_everything(1) img = Image.open(f"{TESTS_FOLDER}/data/girl_with_a_pearl_earring.jpg") caption = generate_caption(img) assert caption in { "a painting of a girl with a pearl earring wearing a yellow dress and a pearl earring in her ear and a black background", "a painting of a girl with a pearl ear wearing a yellow dress and a pearl earring on her left ear and a black background", + "a painting of a woman with a pearl ear wearing an ornament pearl earring and wearing an orange, white, blue and yellow dress", }