Skip to content
This repository was archived by the owner on Oct 11, 2024. It is now read-only.

Commit 81ddde3

Browse files
ywang96Robert Shaw
authored andcommitted
[CI/Build] Fix Args for _get_logits_warper in Sampler Test (vllm-project#5922)
1 parent 6e594ee commit 81ddde3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/samplers/test_sampler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -593,7 +593,7 @@ def test_sampler_top_k_top_p(seed: int, device: str):
593593
generation_config = GenerationConfig(top_k=top_k,
594594
top_p=top_p,
595595
do_sample=True)
596-
warpers = generation_model._get_logits_warper(generation_config)
596+
warpers = generation_model._get_logits_warper(generation_config, device)
597597
assert len(warpers) == 2 # top_p and top_k
598598

599599
seq_group_metadata_list: List[SequenceGroupMetadata] = []

0 commit comments

Comments
 (0)