diff --git a/.buildkite/cuda/test-merge.yml b/.buildkite/cuda/test-merge.yml index e76487cce..021f5d678 100644 --- a/.buildkite/cuda/test-merge.yml +++ b/.buildkite/cuda/test-merge.yml @@ -18,7 +18,7 @@ steps: mirror_hardwares: l4_1 - label: "Simple · Diffusion Test" - timeout_in_minutes: 20 + timeout_in_minutes: 30 commands: - "pytest -sv tests/diffusion -m 'core_model and cpu'" mirror_hardwares: l4_1 diff --git a/.buildkite/cuda/test-ready.yml b/.buildkite/cuda/test-ready.yml index 771554c0f..5fe28722f 100644 --- a/.buildkite/cuda/test-ready.yml +++ b/.buildkite/cuda/test-ready.yml @@ -17,7 +17,7 @@ steps: mirror_hardwares: l4_1 - label: "Simple · Diffusion Test" - timeout_in_minutes: 20 + timeout_in_minutes: 30 commands: - "pytest -sv tests/diffusion -m 'core_model and cpu'" mirror_hardwares: l4_1