[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

@ -121,7 +121,7 @@ def test_cumem_with_cudagraph():
"model, use_v1",
[
# sleep mode with safetensors
(f"{MODEL_WEIGHTS_S3_BUCKET}/Llama-3.2-1B", True),
(f"{MODEL_WEIGHTS_S3_BUCKET}/meta-llama/Llama-3.2-1B", True),
# sleep mode with pytorch checkpoint
("facebook/opt-125m", False),
])