File tree 1 file changed +4
-0
lines changed
1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -472,12 +472,14 @@ steps:
472
472
473
473
- label : Language Models Test (Standard)
474
474
mirror_hardwares : [amdexperimental]
475
+ torch_nightly : true
475
476
source_file_dependencies :
476
477
- vllm/
477
478
- tests/models/language
478
479
commands :
479
480
# Install causal-conv1d for plamo2 models here, as it is not compatible with pip-compile.
480
481
- pip install 'git+https://github.com/Dao-AILab/causal-conv1d@v1.5.0.post8'
482
+ - pip freeze | grep -E 'torch'
481
483
- pytest -v -s models/language -m core_model
482
484
483
485
- label : Language Models Test (Extended)
@@ -493,11 +495,13 @@ steps:
493
495
494
496
- label : Multi-Modal Models Test (Standard)
495
497
mirror_hardwares : [amdexperimental]
498
+ torch_nightly : true
496
499
source_file_dependencies :
497
500
- vllm/
498
501
- tests/models/multimodal
499
502
commands :
500
503
- pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
504
+ - pip freeze | grep -E 'torch'
501
505
- pytest -v -s models/multimodal/processing
502
506
- pytest -v -s --ignore models/multimodal/generation/test_whisper.py models/multimodal -m core_model
503
507
- cd .. && pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model # Otherwise, mp_method="spawn" doesn't work
You can’t perform that action at this time.
0 commit comments