tests/fix: check flux n_iter same as other n_iter tests

The first in a batch should match the generation with a same payload but with `n: 1`.
Haidra-Org · Oct 3, 2024 · b4f87ad · b4f87ad
1 parent 8b7eea1
commit b4f87ad
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 36 deletions.
diff --git a/images_expected/flux_schnell_fp8_text_to_image.png b/images_expected/flux_schnell_fp8_text_to_image.png
diff --git a/tests/test_horde_inference_flux.py b/tests/test_horde_inference_flux.py
@@ -11,42 +11,42 @@
 
 class TestHordeInferenceFlux:
 
-    # @pytest.mark.default_flux1_model
-    # def test_flux_dev_fp8_text_to_image(
-    #     self,
-    #     hordelib_instance: HordeLib,
-    #     flux1_dev_fp8_base_model_name: str,
-    # ):
-    #     data = {
-    #         "sampler_name": "k_euler",
-    #         "cfg_scale": 1,
-    #         "denoising_strength": 1.0,
-    #         "seed": 13122,
-    #         "height": 1024,
-    #         "width": 1024,
-    #         "karras": False,
-    #         "tiling": False,
-    #         "hires_fix": False,
-    #         "clip_skip": 1,
-    #         "control_type": None,
-    #         "image_is_control": False,
-    #         "return_control_map": False,
-    #         "prompt": 'a steampunk text that says "Horde Engine" floating',
-    #         "ddim_steps": 20,
-    #         "n_iter": 1,
-    #         "model": flux1_dev_fp8_base_model_name,
-    #     }
-    #     pil_image = hordelib_instance.basic_inference_single_image(data).image
-    #     assert pil_image is not None
-    #     assert isinstance(pil_image, Image.Image)
+    @pytest.mark.default_flux1_model
+    def test_flux_schnell_fp8_text_to_image(
+        self,
+        hordelib_instance: HordeLib,
+        flux1_schnell_fp8_base_model_name: str,
+    ):
+        data = {
+            "sampler_name": "k_euler",
+            "cfg_scale": 1,
+            "denoising_strength": 1.0,
+            "seed": 13122,
+            "height": 1024,
+            "width": 1024,
+            "karras": False,
+            "tiling": False,
+            "hires_fix": False,
+            "clip_skip": 1,
+            "control_type": None,
+            "image_is_control": False,
+            "return_control_map": False,
+            "prompt": 'a steampunk text that says "Horde Engine" floating',
+            "ddim_steps": 4,
+            "n_iter": 1,
+            "model": flux1_schnell_fp8_base_model_name,
+        }
+        pil_image = hordelib_instance.basic_inference_single_image(data).image
+        assert pil_image is not None
+        assert isinstance(pil_image, Image.Image)
 
-    #     img_filename = "flux_dev_fp8_text_to_image.png"
-    #     pil_image.save(f"images/{img_filename}", quality=100)
+        img_filename = "flux_schnell_fp8_text_to_image.png"
+        pil_image.save(f"images/{img_filename}", quality=100)
 
-    #     assert check_single_inference_image_similarity(
-    #         f"images_expected/{img_filename}",
-    #         pil_image,
-    #     )
+        assert check_single_inference_image_similarity(
+            f"images_expected/{img_filename}",
+            pil_image,
+        )
 
     @pytest.mark.default_flux1_model
     def test_flux_schnell_fp8_text_to_image_n_iter(
@@ -91,8 +91,8 @@ def test_flux_schnell_fp8_text_to_image_n_iter(
             img_pairs_to_check.append((f"images_expected/{img_filename}", image_result.image))
 
         assert check_single_inference_image_similarity(
-            "images_expected/text_to_image.png",
-            "images/text_to_image_n_iter_0.png",
+            "images_expected/flux_schnell_fp8_text_to_image.png",
+            "images/flux_schnell_fp8_text_to_image_n_iter_0.png",
         )
 
         assert check_list_inference_images_similarity(img_pairs_to_check)