add timm extra dependency

.github/workflows/test_cpu_onnxruntime.yaml

@@ -26,7 +26,7 @@ jobs:
       - name: Install requirements
         run: |
           pip install --upgrade pip
-          pip install -e .[test,onnxruntime,diffusers]
+          pip install -e .[test,onnxruntime,diffusers,timm]
 
       - name: Run tests
         run: |

.github/workflows/test_cpu_pytorch.yaml

@@ -26,7 +26,7 @@ jobs:
       - name: Install requirements
         run: |
           pip install --upgrade pip
-          pip install -e .[test,diffusers]
+          pip install -e .[test,diffusers,timm]
 
       - name: Run tests
         run: |

.github/workflows/test_cuda_onnxruntime_inference.yaml

@@ -40,4 +40,4 @@ jobs:
           --workdir /workspace/optimum-benchmark
           --gpus '"device=0,1"'
           opt-bench-cuda:11.8.0
-          -c "pip install -e .[test,onnxruntime-gpu,diffusers] && pytest -k 'cuda and onnxruntime and inference' -x"
+          -c "pip install -e .[test,onnxruntime-gpu] && pytest -k 'cuda and onnxruntime and inference' -x"

.github/workflows/test_cuda_pytorch.yaml

@@ -50,4 +50,4 @@ jobs:
           --gpus '"device=0,1"'
           --entrypoint /bin/bash
           opt-bench-cuda:${{ matrix.image.cuda_version }}
-          -c "pip install -e .[test,peft,diffusers,deepspeed] && pytest -k 'cuda and pytorch' -x"
+          -c "pip install -e .[test,peft,deepspeed] && pytest -k 'cuda and pytorch' -x"

.github/workflows/test_rocm_pytorch.yaml

@@ -53,4 +53,4 @@ jobs:
           --device /dev/dri/renderD129
           --entrypoint /bin/bash
           opt-bench-rocm:${{ matrix.image.rocm_version }}
-          -c "pip install -e .[test,peft,diffusers,deepspeed] && pytest -k 'cuda and pytorch' -x"
+          -c "pip install -e .[test,peft,deepspeed] && pytest -k 'cuda and pytorch' -x"

.github/workflows/test_tensorrt_onnxruntime_inference.yaml

@@ -40,4 +40,4 @@ jobs:
           --gpus '"device=0,1"'
           --entrypoint /bin/bash
           opt-bench-tensorrt:22.12
-          -c "pip install -e .[test,onnxruntime-gpu,diffusers] && pytest -k 'tensorrt and onnxruntime and inference' -x"
+          -c "pip install -e .[test,onnxruntime-gpu] && pytest -k 'tensorrt and onnxruntime and inference' -x"

setup.py

@@ -63,6 +63,7 @@
     # specific settings
     "deepspeed": ["deepspeed"],
     "diffusers": ["diffusers"],
+    "timm": ["timm"],
     "peft": ["peft"],
 }