[VLM][Bugfix] Make sure that multi_modal_kwargs is broadcasted properly (#5880)

Signed-off-by: Xiaowei Jiang <xwjiang2010@gmail.com>
2024-06-27 00:15:24 -07:00
parent 6eabc6cb0e
commit d12af207d2
3 changed files with 81 additions and 9 deletions
--- a/.buildkite/test-pipeline.yaml
+++ b/.buildkite/test-pipeline.yaml
@@ -27,7 +27,9 @@ steps:

 - label: Core Test
  mirror_hardwares: [amd]
-  command: pytest -v -s core
+  commands: 
+  - pytest -v -s core
+  - pytest -v -s distributed/test_parallel_state.py

 - label: Distributed Comm Ops Test
  #mirror_hardwares: [amd]