[CI] Upgrade trasnformers version (#6307)
Upgrade transformers to >=4.56.4
- vLLM version: v0.14.1
- vLLM main:
dc917cceb8
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
@@ -20,7 +20,6 @@
|
||||
import os
|
||||
|
||||
import pytest
|
||||
from modelscope import snapshot_download # type: ignore
|
||||
from vllm import SamplingParams
|
||||
from vllm.assets.audio import AudioAsset
|
||||
|
||||
@@ -46,7 +45,7 @@ def test_minicpm(model) -> None:
|
||||
]
|
||||
max_tokens = 5
|
||||
|
||||
with VllmRunner(snapshot_download(model),
|
||||
with VllmRunner(model,
|
||||
max_model_len=512,
|
||||
gpu_memory_utilization=0.7) as runner:
|
||||
runner.generate_greedy(example_prompts, max_tokens)
|
||||
@@ -61,7 +60,7 @@ def test_whisper(model) -> None:
|
||||
max_tokens=10,
|
||||
stop_token_ids=None)
|
||||
|
||||
with VllmRunner(snapshot_download(model),
|
||||
with VllmRunner(model,
|
||||
max_model_len=448,
|
||||
max_num_seqs=5,
|
||||
dtype="bfloat16",
|
||||
|
||||
Reference in New Issue
Block a user