From 0e56738d57945016e1dfaff743aa0519e954896a Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Sat, 13 Jan 2024 15:37:07 +0100 Subject: [PATCH] add timm extra dependency --- .github/workflows/test_cpu_onnxruntime.yaml | 2 +- .github/workflows/test_cpu_pytorch.yaml | 2 +- .github/workflows/test_cuda_onnxruntime_inference.yaml | 2 +- .github/workflows/test_cuda_pytorch.yaml | 2 +- .github/workflows/test_rocm_pytorch.yaml | 2 +- .github/workflows/test_tensorrt_onnxruntime_inference.yaml | 2 +- setup.py | 1 + 7 files changed, 7 insertions(+), 6 deletions(-) diff --git a/.github/workflows/test_cpu_onnxruntime.yaml b/.github/workflows/test_cpu_onnxruntime.yaml index f578138f..17242b56 100644 --- a/.github/workflows/test_cpu_onnxruntime.yaml +++ b/.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: | diff --git a/.github/workflows/test_cpu_pytorch.yaml b/.github/workflows/test_cpu_pytorch.yaml index 93bc9b8a..8b2f881e 100644 --- a/.github/workflows/test_cpu_pytorch.yaml +++ b/.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: | diff --git a/.github/workflows/test_cuda_onnxruntime_inference.yaml b/.github/workflows/test_cuda_onnxruntime_inference.yaml index fca17e5f..10ae70fc 100644 --- a/.github/workflows/test_cuda_onnxruntime_inference.yaml +++ b/.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" diff --git a/.github/workflows/test_cuda_pytorch.yaml b/.github/workflows/test_cuda_pytorch.yaml index dd845801..8e3216d9 100644 --- a/.github/workflows/test_cuda_pytorch.yaml +++ b/.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" diff --git a/.github/workflows/test_rocm_pytorch.yaml b/.github/workflows/test_rocm_pytorch.yaml index 336af359..cc76e764 100644 --- a/.github/workflows/test_rocm_pytorch.yaml +++ b/.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" diff --git a/.github/workflows/test_tensorrt_onnxruntime_inference.yaml b/.github/workflows/test_tensorrt_onnxruntime_inference.yaml index 5424c19e..9a9d2ef0 100644 --- a/.github/workflows/test_tensorrt_onnxruntime_inference.yaml +++ b/.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" diff --git a/setup.py b/setup.py index 63329634..4d8ed023 100644 --- a/setup.py +++ b/setup.py @@ -63,6 +63,7 @@ # specific settings "deepspeed": ["deepspeed"], "diffusers": ["diffusers"], + "timm": ["timm"], "peft": ["peft"], }