mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-16 09:45:30 +08:00
[CI/Build] do not enforce precompilation on tpu ci tests (#25992)
Signed-off-by: Xiang Si <sixiang@google.com> Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
parent
fac9b430ec
commit
d8b1f9ccc3
@ -123,13 +123,10 @@ def test_structured_output(
|
||||
if current_platform.is_tpu() and speculative_config:
|
||||
pytest.skip("TPU does not support speculative decoding")
|
||||
|
||||
# Don't use eager execution on TPUs because we want to test for no
|
||||
# recompilation at runtime
|
||||
enforce_eager = bool(not current_platform.is_tpu())
|
||||
# Use a single LLM instance for several scenarios to
|
||||
# speed up the test suite.
|
||||
llm = LLM(model=model_name,
|
||||
enforce_eager=enforce_eager,
|
||||
enforce_eager=True,
|
||||
max_model_len=1024,
|
||||
structured_outputs_config=dict(backend=backend,
|
||||
disable_any_whitespace=backend
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user