Skip to content

Commit 9723b2d

Browse files
ywang96jimpang
authored andcommitted
[CI/Build] Fix Args for _get_logits_warper in Sampler Test (vllm-project#5922)
1 parent aa21c3d commit 9723b2d

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
@@ -587,7 +587,7 @@ def test_sampler_top_k_top_p(seed: int, device: str):
587587
generation_config = GenerationConfig(top_k=top_k,
588588
top_p=top_p,
589589
do_sample=True)
590-
warpers = generation_model._get_logits_warper(generation_config)
590+
warpers = generation_model._get_logits_warper(generation_config, device)
591591
assert len(warpers) == 2 # top_p and top_k
592592

593593
seq_group_metadata_list: List[SequenceGroupMetadata] = []

0 commit comments

Comments
 (0)