Skip to content

Commit 2b0db9b

Browse files
authored
Enable standard language model for torhc nightly (#18004)
Signed-off-by: Yang Wang <elainewy@meta.com>
1 parent 195adb4 commit 2b0db9b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -472,12 +472,14 @@ steps:
472472

473473
- label: Language Models Test (Standard)
474474
mirror_hardwares: [amdexperimental]
475+
torch_nightly: true
475476
source_file_dependencies:
476477
- vllm/
477478
- tests/models/language
478479
commands:
479480
# Install causal-conv1d for plamo2 models here, as it is not compatible with pip-compile.
480481
- pip install 'git+https://github.com/Dao-AILab/causal-conv1d@v1.5.0.post8'
482+
- pip freeze | grep -E 'torch'
481483
- pytest -v -s models/language -m core_model
482484

483485
- label: Language Models Test (Extended)
@@ -493,11 +495,13 @@ steps:
493495

494496
- label: Multi-Modal Models Test (Standard)
495497
mirror_hardwares: [amdexperimental]
498+
torch_nightly: true
496499
source_file_dependencies:
497500
- vllm/
498501
- tests/models/multimodal
499502
commands:
500503
- pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
504+
- pip freeze | grep -E 'torch'
501505
- pytest -v -s models/multimodal/processing
502506
- pytest -v -s --ignore models/multimodal/generation/test_whisper.py models/multimodal -m core_model
503507
- cd .. && pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model # Otherwise, mp_method="spawn" doesn't work

0 commit comments

Comments
 (0)