up

Author: sayakpaul

tests/pipelines/test_pipelines.py

@@ -28,7 +28,6 @@
 
 import numpy as np
 import PIL.Image
-import pytest
 import requests_mock
 import safetensors.torch
 import torch
@@ -63,7 +62,7 @@
 )
 from diffusers.pipelines.pipeline_utils import _get_pipeline_class
 from diffusers.schedulers.scheduling_utils import SCHEDULER_CONFIG_NAME
-from diffusers.utils import CONFIG_NAME, WEIGHTS_NAME, is_transformers_version
+from diffusers.utils import CONFIG_NAME, WEIGHTS_NAME
 from diffusers.utils.torch_utils import is_compiled_module
 
 from ..testing_utils import (
@@ -582,7 +581,6 @@ def test_download_variants_with_sharded_checkpoints(self):
                     assert not any(f.endswith(unexpected_ext) for f in files)
                     assert all(variant in f for f in model_files if f.endswith(model_ext) and variant is not None)
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_download_legacy_variants_with_sharded_ckpts_raises_warning(self):
         repo_id = "hf-internal-testing/tiny-stable-diffusion-pipe-variants-all-kinds"
         logger = logging.get_logger("diffusers.pipelines.pipeline_utils")
@@ -600,7 +598,6 @@ def test_download_legacy_variants_with_sharded_ckpts_raises_warning(self):
                 )
         assert deprecated_warning_msg in str(cap_logger), "Deprecation warning not found in logs"
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_download_safetensors_only_variant_exists_for_model(self):
         variant = None
         use_safetensors = True
@@ -630,7 +627,7 @@ def test_download_safetensors_only_variant_exists_for_model(self):
             # https://huggingface.co/hf-internal-testing/stable-diffusion-broken-variants/tree/main/unet
             assert len(files) == 15, f"We should only download 15 files, not {len(files)}"
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
+    #
     def test_download_bin_only_variant_exists_for_model(self):
         variant = None
         use_safetensors = False
@@ -660,7 +657,6 @@ def test_download_bin_only_variant_exists_for_model(self):
             # https://huggingface.co/hf-internal-testing/stable-diffusion-broken-variants/tree/main/unet
             assert len(files) == 15, f"We should only download 15 files, not {len(files)}"
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_download_safetensors_variant_does_not_exist_for_model(self):
         variant = "no_ema"
         use_safetensors = True
@@ -677,7 +673,6 @@ def test_download_safetensors_variant_does_not_exist_for_model(self):
 
             assert "Could not find the necessary `safetensors` weights" in str(error_context.exception)
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_download_bin_variant_does_not_exist_for_model(self):
         variant = "no_ema"
         use_safetensors = False
@@ -693,7 +688,6 @@ def test_download_bin_variant_does_not_exist_for_model(self):
                 )
             assert "Error no file name" in str(error_context.exception)
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_local_save_load_index(self):
         prompt = "hello"
         for variant in [None, "fp16"]:
@@ -1588,7 +1582,6 @@ def test_save_safe_serialization(self):
             assert pipeline.scheduler is not None
             assert pipeline.feature_extractor is not None
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_no_pytorch_download_when_doing_safetensors(self):
         # by default we don't download
         with tempfile.TemporaryDirectory() as tmpdirname:
@@ -1608,7 +1601,6 @@ def test_no_pytorch_download_when_doing_safetensors(self):
             # pytorch does not
             assert not os.path.exists(os.path.join(path, "diffusion_pytorch_model.bin"))
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_no_safetensors_download_when_doing_pytorch(self):
         use_safetensors = False
 
@@ -1894,7 +1886,6 @@ def test_dduf_raises_error_with_connected_pipeline(self):
                 "DDUF/tiny-flux-dev-pipe-dduf", dduf_file="fluxpipeline.dduf", load_connected_pipeline=True
             )
 
-    @pytest.mark.xfail(condition=is_transformers_version(">", "4.56.2"), reason="Some import error", strict=True)
     def test_wrong_model(self):
         tokenizer = CLIPTokenizer.from_pretrained("hf-internal-testing/tiny-random-clip")
         with self.assertRaises(ValueError) as error_context: