raise error when incompatible diffusers andtransformers and don't tes…

…t difusers with transformers 4.36

.github/workflows/test_onnxruntime.yml

@@ -38,14 +38,18 @@ jobs:
       - name: Setup Python
         uses: actions/setup-python@v5
         with:
-          python-version: '3.9'
+          python-version: "3.9"
 
       - name: Install dependencies
         run: |
           pip install --upgrade pip
           pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
           pip install .[tests,onnxruntime,diffusers]
 
+      - name: Remove Diffusers if transformers 4.36
+        if: matrix.transformers-version == '4.36.*'
+        run: pip uninstall -y diffusers
+
       - name: Install transformers ${{ matrix.transformers-version }}
         if: ${{ matrix.transformers-version != 'latest' }}
         run: pip install transformers==${{ matrix.transformers-version }}

optimum/exporters/tasks.py

@@ -29,7 +29,12 @@
 from transformers import AutoConfig, PretrainedConfig, is_tf_available, is_torch_available
 from transformers.utils import SAFE_WEIGHTS_NAME, TF2_WEIGHTS_NAME, WEIGHTS_NAME, logging
 
-from ..utils.import_utils import is_diffusers_available, is_onnx_available
+from ..utils.import_utils import (
+    is_diffusers_available,
+    is_diffusers_version,
+    is_onnx_available,
+    is_transformers_version,
+)
 
 
 if TYPE_CHECKING:
@@ -51,6 +56,12 @@
     from transformers import TFPreTrainedModel
 
 if is_diffusers_available():
+    if is_diffusers_version(">=", "0.32.0") and is_transformers_version("<", "4.41.2"):
+        raise ImportError(
+            "The current version of `diffusers` requires `transformers>=4.41.2`."
+            " Please upgrade to the latest version of Transformers."
+        )
+
     from diffusers import DiffusionPipeline
     from diffusers.pipelines.auto_pipeline import (
         AUTO_IMAGE2IMAGE_PIPELINES_MAPPING,