File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff 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/[email protected] ' 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
You can’t perform that action at this time.
0 commit comments