Skip to content

Commit 46358ed

Browse files
committed
fix test
Signed-off-by: jesse <[email protected]>
1 parent b4e4098 commit 46358ed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/e2e/singlecard/test_completion_with_prompt_embeds.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
os.environ["VLLM_USE_MODELSCOPE"] = "True"
2626
os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn"
2727

28-
MODELS = ["meta-llama/Llama-3.2-1B-Instruct"]
28+
MODELS = ["Qwen/Qwen2.5-0.5B-Instruct"]
2929

3030

3131
def get_prompt_embeds(chat, tokenizer, embedding_layer):

0 commit comments

Comments
 (0)