Skip to content

[Test] L4 complete diffusion feature test for Wan2.2 models #2087

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
david6666666 merged 9 commits into from
Mar 26, 2026
+517 −49
Merged

[Test] L4 complete diffusion feature test for Wan2.2 models #2087

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
d4f4ed8
add wan22 l4 test
bjf-frz Mar 23, 2026
3c56f67
modify req logic
bjf-frz Mar 23, 2026
f8ad245
fix process image diffusion
bjf-frz Mar 24, 2026
0dad17e
Merge branch 'main' into wan22_l4_test
bjf-frz Mar 24, 2026
a7a359e
prolong duration of nightly test with h100
bjf-frz Mar 24, 2026
4aa563d
modify hunyuan-video l4 model test
bjf-frz Mar 25, 2026
42864f8
Merge branch 'main' into wan22_l4_test
bjf-frz Mar 25, 2026
8d25f4a
add l4 label for wan22
bjf-frz Mar 26, 2026
60dcdc1
Merge branch 'main' into wan22_l4_test
david6666666 Mar 26, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
41 changes: 41 additions & 0 deletions .buildkite/test-nightly.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -111,6 +111,47 @@ steps:
path: /mnt/hf-cache
type: DirectoryOrCreate

- label: ":full_moon: Diffusion Model Wan22 completed Test with H100"
timeout_in_minutes: 45
depends_on: upload-nightly-pipeline
if: build.env("NIGHTLY") == "1" || build.pull_request.labels includes "nightly-test"
commands:
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
- pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -m "advanced_model" --run-level "advanced_model"
Copy link
Copy Markdown
Collaborator

@yenuo26 yenuo26 Mar 26, 2026 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think just modifying here maybe cause both "Diffusion Model Wan22 completed Test with H100" and "Diffusion Model Test with H100" to run this test case.

Copy link
Copy Markdown
Contributor Author

@bjf-frz bjf-frz Mar 26, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I removed the diffusion mark in the wan22 test script, it won't run in Diffusion Model Test with H100.

Copy link
Copy Markdown
Collaborator

@yenuo26 yenuo26 Mar 26, 2026 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe you can change to use pytest --ignore in subsequent PR? I think not adding the diffusion tag may cause some statistical issues later.

Copy link
Copy Markdown
Contributor Author

@bjf-frz bjf-frz Mar 26, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ok

agents:
queue: "mithril-h100-pool"
plugins:
- kubernetes:
podSpec:
containers:
- image: 936637512419.dkr.ecr.us-west-2.amazonaws.com/vllm-ci-pull-through-cache/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
resources:
limits:
nvidia.com/gpu: 2
volumeMounts:
- name: devshm
mountPath: /dev/shm
- name: hf-cache
mountPath: /root/.cache/huggingface
env:
- name: HF_HOME
value: /root/.cache/huggingface
- name: HF_TOKEN
valueFrom:
secretKeyRef:
name: hf-token-secret
key: token
nodeSelector:
node.kubernetes.io/instance-type: gpu-h100-sxm
volumes:
- name: devshm
emptyDir:
medium: Memory
- name: hf-cache
hostPath:
path: /mnt/hf-cache
type: DirectoryOrCreate

- label: ":full_moon: Diffusion Model Test with L4"
timeout_in_minutes: 60
depends_on: upload-nightly-pipeline
Expand Down
Loading
Loading