added sigmas test

Author: SahilCarterr

tests/pipelines/flux/test_pipeline_flux.py

@@ -205,6 +205,17 @@ def test_flux_image_output_shape(self):
             output_height, output_width, _ = image.shape
             assert (output_height, output_width) == (expected_height, expected_width)
 
+    def test_sigmas(self):
+        pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
+        inputs = self.get_dummy_inputs(torch_device)
+
+        output_without_sigmas = pipe(**inputs).images[0]
+        inputs = self.get_dummy_inputs(torch_device)
+        inputs["sigmas"] = [0.25]
+        output_with_sigmas = pipe(**inputs).images[0]
+        max_diff = np.abs(output_without_sigmas - output_with_sigmas).max()
+        assert max_diff > 1e-4
+
 
 @slow
 @require_big_gpu_with_torch_cuda

tests/pipelines/flux/test_pipeline_flux_control.py

@@ -201,3 +201,14 @@ def test_flux_image_output_shape(self):
             image = pipe(**inputs).images[0]
             output_height, output_width, _ = image.shape
             assert (output_height, output_width) == (expected_height, expected_width)
+
+    def test_sigmas(self):
+        pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
+        inputs = self.get_dummy_inputs(torch_device)
+
+        output_without_sigmas = pipe(**inputs).images[0]
+        inputs = self.get_dummy_inputs(torch_device)
+        inputs["sigmas"] = [0.25]
+        output_with_sigmas = pipe(**inputs).images[0]
+        max_diff = np.abs(output_without_sigmas - output_with_sigmas).max()
+        assert max_diff > 1e-4

tests/pipelines/flux/test_pipeline_flux_control_img2img.py

@@ -166,3 +166,14 @@ def test_flux_image_output_shape(self):
             image = pipe(**inputs).images[0]
             output_height, output_width, _ = image.shape
             assert (output_height, output_width) == (expected_height, expected_width)
+
+    def test_sigmas(self):
+        pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
+        inputs = self.get_dummy_inputs(torch_device)
+
+        output_without_sigmas = pipe(**inputs).images[0]
+        inputs = self.get_dummy_inputs(torch_device)
+        inputs["sigmas"] = [0.25]
+        output_with_sigmas = pipe(**inputs).images[0]
+        max_diff = np.abs(output_without_sigmas - output_with_sigmas).max()
+        assert max_diff > 1e-4

tests/pipelines/flux/test_pipeline_flux_fill.py

@@ -166,3 +166,14 @@ def test_flux_image_output_shape(self):
 
     def test_inference_batch_single_identical(self):
         self._test_inference_batch_single_identical(expected_max_diff=1e-3)
+
+    def test_sigmas(self):
+        pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
+        inputs = self.get_dummy_inputs(torch_device)
+
+        output_without_sigmas = pipe(**inputs).images[0]
+        inputs = self.get_dummy_inputs(torch_device)
+        inputs["sigmas"] = [0.25]
+        output_with_sigmas = pipe(**inputs).images[0]
+        max_diff = np.abs(output_without_sigmas - output_with_sigmas).max()
+        assert max_diff > 1e-4

tests/pipelines/flux/test_pipeline_flux_img2img.py

@@ -161,3 +161,14 @@ def test_flux_image_output_shape(self):
             image = pipe(**inputs).images[0]
             output_height, output_width, _ = image.shape
             assert (output_height, output_width) == (expected_height, expected_width)
+
+    def test_sigmas(self):
+        pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
+        inputs = self.get_dummy_inputs(torch_device)
+
+        output_without_sigmas = pipe(**inputs).images[0]
+        inputs = self.get_dummy_inputs(torch_device)
+        inputs["sigmas"] = [0.25]
+        output_with_sigmas = pipe(**inputs).images[0]
+        max_diff = np.abs(output_without_sigmas - output_with_sigmas).max()
+        assert max_diff > 1e-4

tests/pipelines/flux/test_pipeline_flux_inpaint.py

@@ -163,3 +163,14 @@ def test_flux_image_output_shape(self):
             image = pipe(**inputs).images[0]
             output_height, output_width, _ = image.shape
             assert (output_height, output_width) == (expected_height, expected_width)
+
+    def test_sigmas(self):
+        pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
+        inputs = self.get_dummy_inputs(torch_device)
+
+        output_without_sigmas = pipe(**inputs).images[0]
+        inputs = self.get_dummy_inputs(torch_device)
+        inputs["sigmas"] = [0.25]
+        output_with_sigmas = pipe(**inputs).images[0]
+        max_diff = np.abs(output_without_sigmas - output_with_sigmas).max()
+        assert max_diff > 1e-4