Weiming Liaoandwangyu
7aa5c9a090
Signed-off-by: Weiming Liao <liaowm5@gmail.com> Co-authored-by: wangyu <53896905+yenuo26@users.noreply.github.com>
53 lines
2.1 KiB
YAML
53 lines
2.1 KiB
YAML
steps:
|
|
- group: ":card_index_dividers: Diffusion X2V Model Test"
|
|
key: nightly-diffusion-x2v-group
|
|
depends_on: upload-nightly-pipeline
|
|
if: >-
|
|
build.env("NIGHTLY") == "1" ||
|
|
build.pull_request.labels includes "nightly-test" ||
|
|
build.pull_request.labels includes "diffusion-x2v-test"
|
|
steps:
|
|
- label: ":full_moon: Diffusion X2V · Function Test"
|
|
agents:
|
|
queue: "ascend-a2b3"
|
|
resource_class: "npu-4"
|
|
image: "swr.cn-southwest-2.myhuaweicloud.com/modelfoundry/vllm-omni-ci-npu:${BUILDKITE_COMMIT}"
|
|
plugins:
|
|
- kubernetes:
|
|
podSpecPatch:
|
|
imagePullSecrets:
|
|
- name: swr-secret
|
|
env:
|
|
VLLM_WORKER_MULTIPROC_METHOD: spawn
|
|
HF_TOKEN: "${HF_TOKEN}"
|
|
commands:
|
|
- pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -k npu_i2v --run-level "full_model"
|
|
- group: ":card_index_dividers: TTS Model Test"
|
|
key: nightly-tts-test-group
|
|
depends_on: upload-nightly-pipeline
|
|
if: build.env("NIGHTLY") == "1" || build.pull_request.labels includes "nightly-test" || build.pull_request.labels includes "tts-test"
|
|
steps:
|
|
- label: ":full_moon: TTS · Perf Test"
|
|
agents:
|
|
queue: "ascend-a3"
|
|
resource_class: "npu-2"
|
|
image: "swr.cn-southwest-2.myhuaweicloud.com/modelfoundry/vllm-omni-ci-npu:${BUILDKITE_COMMIT}-a3"
|
|
plugins:
|
|
- kubernetes:
|
|
podSpecPatch:
|
|
imagePullSecrets:
|
|
- name: swr-secret
|
|
env:
|
|
VLLM_WORKER_MULTIPROC_METHOD: spawn
|
|
VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
|
|
HF_TOKEN: "${HF_TOKEN}"
|
|
commands:
|
|
- export BENCHMARK_DIR=tests/dfx/perf/results
|
|
- export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
|
|
- |
|
|
set +e
|
|
pytest -s -v tests/dfx/perf/scripts/run_benchmark.py -m npu --test-config-file tests/dfx/perf/tests/test_tts.json
|
|
EXIT=$$?
|
|
buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
|
|
exit $$EXIT
|