File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
.buildkite/scripts/hardware_ci Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -32,9 +32,12 @@ function cpu_tests() {
3232 set -e
3333 pip install pytest pytest-asyncio einops peft Pillow soundfile transformers_stream_generator matplotlib
3434 pip install sentence-transformers datamodel_code_generator
35- pytest -v -s tests/models/embedding/language/test_cls_models.py::test_classification_models[float-jason9693/Qwen2.5-1.5B-apeach]
36- pytest -v -s tests/models/embedding/language/test_embedding.py::test_models[half-BAAI/bge-base-en-v1.5]
37- pytest -v -s tests/models/encoder_decoder/language -m cpu_model"
35+ pytest -v -s tests/models/language/generation/test_bart.py -m cpu_model
36+ pytest -v -s tests/models/language/generation/test_common.py::test_models[False-5-32-openai-community/gpt2]
37+ pytest -v -s tests/models/language/generation/test_common.py::test_models[False-5-32-facebook/opt-125m]
38+ pytest -v -s tests/models/language/generation/test_common.py::test_models[False-5-32-google/gemma-1.1-2b-it]
39+ pytest -v -s tests/models/language/pooling/test_classification.py::test_models[float-jason9693/Qwen2.5-1.5B-apeach]
40+ pytest -v -s tests/models/language/pooling/test_embedding.py::test_models[half-BAAI/bge-base-en-v1.5]"
3841}
3942
4043# All of CPU tests are expected to be finished less than 40 mins.
You can’t perform that action at this time.
0 commit comments