[ci] Use env var to control whether to use S3 bucket in CI (#13634)

2025-02-22 19:19:45 -08:00
parent 322d2a27d6
commit 2c5e637b57
30 changed files with 222 additions and 231 deletions
--- a/tests/worker/test_swap.py
+++ b/tests/worker/test_swap.py
@ -10,7 +10,7 @@ from vllm.worker.worker import Worker

 def test_swap() -> None:
    # Configure the engine.
-    engine_args = EngineArgs(model="s3://vllm-ci-model-weights/distilgpt2",
+    engine_args = EngineArgs(model="distilbert/distilgpt2",
                             dtype="half",
                             load_format="dummy")
    engine_config = engine_args.create_engine_config()