diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index 14eb4ec21b44..aa3ec892bb99 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -841,6 +841,7 @@ steps: no_gpu: true commands: - "pip install git+https://github.com/TIGER-AI-Lab/Mantis.git || echo 'Mantis installation skipped (decord not available on CPU-only environment)'" + - pip install timm==1.0.22 - pytest -v -s models/multimodal/processing --ignore models/multimodal/processing/test_tensor_schema.py - label: Multi-Modal Processor Test diff --git a/.buildkite/test_areas/models_multimodal.yaml b/.buildkite/test_areas/models_multimodal.yaml index fc24068c20a4..8ab9efcc9162 100644 --- a/.buildkite/test_areas/models_multimodal.yaml +++ b/.buildkite/test_areas/models_multimodal.yaml @@ -21,6 +21,7 @@ steps: no_gpu: true commands: - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git + - pip install timm==1.0.22 - pytest -v -s models/multimodal/processing --ignore models/multimodal/processing/test_tensor_schema.py - label: Multi-Modal Processor # 44min