[2/n][ci] S3: Use full model path (#13564)

Signed-off-by: <>
This commit is contained in:
Kevin H. Luu
2025-02-20 01:20:15 -08:00
committed by GitHub
parent aa1e62d0db
commit a64a84433d
8 changed files with 26 additions and 19 deletions

View File

@@ -9,7 +9,8 @@ from vllm.sampling_params import SamplingParams
from ..conftest import MODEL_WEIGHTS_S3_BUCKET
@pytest.mark.parametrize("model", [f"{MODEL_WEIGHTS_S3_BUCKET}/distilgpt2"])
@pytest.mark.parametrize("model",
[f"{MODEL_WEIGHTS_S3_BUCKET}/distilbert/distilgpt2"])
def test_skip_tokenizer_initialization(model: str):
# This test checks if the flag skip_tokenizer_init skips the initialization
# of tokenizer and detokenizer. The generated output is expected to contain