[ci] Use env var to control whether to use S3 bucket in CI (#13634)
This commit is contained in:
@@ -21,10 +21,8 @@ from vllm.lora.request import LoRARequest
|
||||
from vllm.usage.usage_lib import UsageContext
|
||||
from vllm.utils import FlexibleArgumentParser
|
||||
|
||||
MODEL = "s3://vllm-ci-model-weights/gemma-1.1-2b-it"
|
||||
ENGINE_ARGS = AsyncEngineArgs(model=MODEL,
|
||||
load_format="runai_streamer",
|
||||
enforce_eager=True)
|
||||
MODEL = "google/gemma-1.1-2b-it"
|
||||
ENGINE_ARGS = AsyncEngineArgs(model=MODEL, enforce_eager=True)
|
||||
RAISED_ERROR = KeyError
|
||||
RAISED_VALUE = "foo"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user