"server/vscode:/vscode.git/clone" did not exist on "8ad60b752fdf678153e7685d9f0e30e88a60cd42"
Unverified Commit 428b988c authored by Andreas Karatzas's avatar Andreas Karatzas Committed by GitHub
Browse files

[ROCm][CI] Fix `trust_remote_code` AttributeError in EAGLE3 acceptance length test (#40306)


Signed-off-by: default avatarAndreas Karatzas <akaratza@amd.com>
parent e54894fc
......@@ -165,6 +165,7 @@ def get_mt_bench_prompts(
no_stream=True,
disable_shuffle=False,
skip_chat_template=False,
trust_remote_code=False,
)
samples = get_samples(args, tokenizer)
prompt_ids = [
......@@ -210,8 +211,8 @@ def extract_acceptance_metrics(metrics, num_spec_tokens: int) -> dict:
@large_gpu_mark(min_gb=40)
@pytest.mark.skipif(
not current_platform.is_cuda(),
reason="This test is only supported on CUDA platform.",
not current_platform.is_cuda_alike(),
reason="This test is only supported on CUDA-alike platforms.",
)
@pytest.mark.parametrize(
"model_config",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment