test(diffusers): disable flux tests

These tests hang with AWS Neuron SDK 2.26

Author: dacorvo

tests/inference/diffusers/test_export_cli.py

@@ -16,10 +16,12 @@
 import tempfile
 import unittest
 
+import pytest
 from optimum.utils import logging
 
 from optimum.exporters.neuron.model_configs import *  # noqa: F403
 from optimum.neuron.utils.testing_utils import is_inferentia_test, requires_neuronx
+from optimum.neuron.version import __sdk_version__ as sdk_version
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
@@ -87,6 +89,7 @@ def test_pixart(self):
                     check=True,
                 )
 
+    @pytest.mark.skipif(sdk_version == "2.26.0", reason="This test hangs with SDK 2.26.0")
     @requires_neuronx
     def test_flux_tp2(self):
         model_ids = ["hf-internal-testing/tiny-flux-pipe-gated-silu"]

tests/inference/diffusers/test_flux.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 
 import PIL
+import pytest
 import torch
 from diffusers.utils import load_image
 from optimum.utils.testing_utils import require_diffusers
@@ -26,8 +27,10 @@
     NeuronModelVaeEncoder,
 )
 from optimum.neuron.utils.testing_utils import is_inferentia_test, requires_neuronx
+from optimum.neuron.version import __sdk_version__ as sdk_version
 
 
+@pytest.mark.skipif(sdk_version == "2.26.0", reason="This test hangs with SDK 2.26.0")
 @is_inferentia_test
 @requires_neuronx
 @require_diffusers
@@ -47,6 +50,7 @@ def test_flux_txt2img(neuron_flux_tp2_path):
     assert isinstance(image, PIL.Image.Image)
 
 
+@pytest.mark.skipif(sdk_version == "2.26.0", reason="This test hangs with SDK 2.26.0")
 @is_inferentia_test
 @requires_neuronx
 @require_diffusers
@@ -68,6 +72,7 @@ def test_flux_inpaint(neuron_flux_tp2_path):
     assert isinstance(image, PIL.Image.Image)
 
 
+@pytest.mark.skipif(sdk_version == "2.26.0", reason="This test hangs with SDK 2.26.0")
 @is_inferentia_test
 @requires_neuronx
 @require_diffusers