xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-03-21 16:01:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

ruff

Browse Source 
Signed-off-by: qli88 <qiang.li2@amd.com>
This commit is contained in:
qli88 2025-12-16 17:20:46 +00:00
parent 23b4fd1715
commit c810903fc9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/samplers/test_no_bad_words.py
View File

@ -96,7 +96,8 @@ class TestTwoTokenBadWord:
def test_two_token_bad_word(self, vllm_runner):
# TODO: Remove once graph mode is fixed for distilbert/distilgpt2 on ROCm.
eager_mode = current_platform.is_rocm()
eager_mode = current_platform.is_rocm()
with vllm_runner(self.MODEL, dtype="half", enforce_eager=eager_mode) as llm:
output_token_ids = self._generate(llm)