File tree Expand file tree Collapse file tree 1 file changed +12
-3
lines changed
Expand file tree Collapse file tree 1 file changed +12
-3
lines changed Original file line number Diff line number Diff line change @@ -486,16 +486,25 @@ steps:
486486 - pip freeze | grep -E 'torch'
487487 - pytest -v -s models/language -m core_model
488488
489- - label : Language Models Test (Extended)
489+ - label : Language Models Test (Extended Generation) # 1hr20min
490490 mirror_hardwares : [amdexperimental]
491491 optional : true
492492 source_file_dependencies :
493493 - vllm/
494- - tests/models/language
494+ - tests/models/language/generation
495495 commands :
496496 # Install causal-conv1d for plamo2 models here, as it is not compatible with pip-compile.
497497 -
pip install 'git+https://github.com/Dao-AILab/[email protected] ' 498- - pytest -v -s models/language -m 'not core_model'
498+ - pytest -v -s models/language/generation -m 'not core_model'
499+
500+ - label : Language Models Test (Extended Pooling) # 36min
501+ mirror_hardwares : [amdexperimental]
502+ optional : true
503+ source_file_dependencies :
504+ - vllm/
505+ - tests/models/language/pooling
506+ commands :
507+ - pytest -v -s models/language/pooling -m 'not core_model'
499508
500509- label : Multi-Modal Models Test (Standard)
501510 mirror_hardwares : [amdexperimental]
You can’t perform that action at this time.
0 commit comments