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:
486
486
- pip freeze | grep -E 'torch'
487
487
- pytest -v -s models/language -m core_model
488
488
489
- - label : Language Models Test (Extended)
489
+ - label : Language Models Test (Extended Generation) # 1hr20min
490
490
mirror_hardwares : [amdexperimental]
491
491
optional : true
492
492
source_file_dependencies :
493
493
- vllm/
494
- - tests/models/language
494
+ - tests/models/language/generation
495
495
commands :
496
496
# Install causal-conv1d for plamo2 models here, as it is not compatible with pip-compile.
497
497
-
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'
499
508
500
509
- label : Multi-Modal Models Test (Standard)
501
510
mirror_hardwares : [amdexperimental]
You can’t perform that action at this time.
0 commit comments