Skip to content

Commit 24830d1

Browse files
committed
Switch to Qwen2.5-3B-Instruct and fix model cache bucket name
- Replace Llama-3.1-8B-Instruct with Qwen/Qwen2.5-3B-Instruct - Rename model cache bucket to aieng-template-cloudrun-vllm - Bump MAX_NUM_SEQS from 16 → 32 (3B fits more concurrency on L4) - Update served model name to qwen2.5-3b-instruct
1 parent 2fe2959 commit 24830d1

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

‎.github/workflows/deploy.yml‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,13 +33,13 @@ env:
3333
VLLM_SERVICE: my-project-vllm # Cloud Run GPU service (vLLM inference)
3434

3535
# ── Model ─────────────────────────────────────────────────────────────────
36-
MODEL_ID: meta-llama/Llama-3.1-8B-Instruct # HuggingFace model ID
37-
MODEL_SERVED_NAME: my-model # Name exposed by the vLLM /v1/models API
38-
MODEL_CACHE_BUCKET: my-project-model-cache # GCS bucket for HF weights
36+
MODEL_ID: Qwen/Qwen2.5-3B-Instruct # HuggingFace model ID
37+
MODEL_SERVED_NAME: qwen2.5-3b-instruct # Name exposed by the vLLM /v1/models API
38+
MODEL_CACHE_BUCKET: aieng-template-cloudrun-vllm # GCS bucket for HF weights
3939

4040
# ── vLLM inference settings ───────────────────────────────────────────────
4141
MAX_MODEL_LEN: 8192 # context window (tokens)
42-
MAX_NUM_SEQS: 16 # max concurrent sequences (L4 sweet spot for 8B)
42+
MAX_NUM_SEQS: 32 # max concurrent sequences (L4 sweet spot for 3B)
4343
DTYPE: bfloat16
4444

4545
# ── Demo app settings ─────────────────────────────────────────────────────

‎Dockerfile.vllm‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,12 +32,12 @@ EXPOSE 8080
3232
#
3333
# For gated models (e.g. Llama 3), pass --hf-token via env var in deploy.yml.
3434
CMD [ \
35-
"meta-llama/Llama-3.1-8B-Instruct", \
36-
"--served-model-name", "my-model", \
35+
"Qwen/Qwen2.5-3B-Instruct", \
36+
"--served-model-name", "qwen2.5-3b-instruct", \
3737
"--port", "8080", \
3838
"--host", "0.0.0.0", \
3939
"--max-model-len", "8192", \
40-
"--max-num-seqs", "16", \
40+
"--max-num-seqs", "32", \
4141
"--dtype", "bfloat16", \
4242
"--enable-chunked-prefill" \
4343
]

0 commit comments

Comments
 (0)