diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml index ca7ed7920..8f942b041 100644 --- a/.buildkite/pipeline.yml +++ b/.buildkite/pipeline.yml @@ -41,23 +41,6 @@ steps: volumes: - "/fsx/hf_cache:/fsx/hf_cache" - - label: "Diffusion Images API LoRA E2E" - timeout_in_minutes: 20 - depends_on: image-build - commands: - - pytest -s -v tests/e2e/online_serving/test_images_generations_lora.py - agents: - queue: "gpu_1_queue" # g6.4xlarge instance on AWS, has 1 L4 GPU - plugins: - - docker#v5.2.0: - image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT - always-pull: true - propagate-environment: true - environment: - - "HF_HOME=/fsx/hf_cache" - volumes: - - "/fsx/hf_cache:/fsx/hf_cache" - - label: "Diffusion Model CPU offloading Test" timeout_in_minutes: 20 depends_on: image-build diff --git a/tests/e2e/online_serving/test_images_generations_lora.py b/tests/e2e/online_serving/test_images_generations_lora.py index 38d3c9b89..e5fabcb81 100644 --- a/tests/e2e/online_serving/test_images_generations_lora.py +++ b/tests/e2e/online_serving/test_images_generations_lora.py @@ -84,7 +84,7 @@ class OmniServer: ) # Wait for server to be ready. - max_wait = 600 + max_wait = 1200 url = f"http://{self.host}:{self.port}/v1/models" start_time = time.time() while time.time() - start_time < max_wait: