[Bugfix] Make MM batching more robust (#33817)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2026-02-06 04:40:58 +08:00
parent 4145e50d85
commit 116880a5a0
13 changed files with 625 additions and 428 deletions
--- a/.buildkite/test_areas/models_basic.yaml
+++ b/.buildkite/test_areas/models_basic.yaml
@@ -33,10 +33,11 @@ steps:
  timeout_in_minutes: 45
  source_file_dependencies:
  - vllm/
+  - tests/models/test_terratorch.py
  - tests/models/test_transformers.py
  - tests/models/test_registry.py
  commands:
-    - pytest -v -s models/test_transformers.py models/test_registry.py
+    - pytest -v -s models/test_terratorch.py models/test_transformers.py models/test_registry.py

 - label: Basic Models Test (Other CPU) # 5min
  depends_on: