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( ...@@ -165,6 +165,7 @@ def get_mt_bench_prompts(
no_stream=True, no_stream=True,
disable_shuffle=False, disable_shuffle=False,
skip_chat_template=False, skip_chat_template=False,
trust_remote_code=False,
) )
samples = get_samples(args, tokenizer) samples = get_samples(args, tokenizer)
prompt_ids = [ prompt_ids = [
...@@ -210,8 +211,8 @@ def extract_acceptance_metrics(metrics, num_spec_tokens: int) -> dict: ...@@ -210,8 +211,8 @@ def extract_acceptance_metrics(metrics, num_spec_tokens: int) -> dict:
@large_gpu_mark(min_gb=40) @large_gpu_mark(min_gb=40)
@pytest.mark.skipif( @pytest.mark.skipif(
not current_platform.is_cuda(), not current_platform.is_cuda_alike(),
reason="This test is only supported on CUDA platform.", reason="This test is only supported on CUDA-alike platforms.",
) )
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model_config", "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