Skip to content

Commit 585b713

Browse files
remove using mocked hf download
Signed-off-by: Govind Ramnarayan <105831528+govind-ramnarayan@users.noreply.github.com>
1 parent f5d89e6 commit 585b713

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unittest/_torch/auto_deploy/unit/singlegpu/test_ad_speculative_decoding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from tensorrt_llm.llmapi import DraftTargetDecodingConfig, KvCacheConfig
2222

2323

24-
@pytest.mark.parametrize("use_hf_speculative_model", [False, True])
24+
@pytest.mark.parametrize("use_hf_speculative_model", [False])
2525
@with_mocked_hf_download
2626
def test_ad_speculative_decoding_smoke(use_hf_speculative_model: bool):
2727
"""Test speculative decoding with AutoDeploy using the build_and_run_ad main()."""

0 commit comments

Comments
 (0)