xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-24 20:31:19 +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

1
tests/samplers/test_no_bad_words.py
View File

@ -96,6 +96,7 @@ class TestTwoTokenBadWord:
def test_two_token_bad_word(self, vllm_runner): def test_two_token_bad_word(self, vllm_runner):
# TODO: Remove once graph mode is fixed for distilbert/distilgpt2 on ROCm. # 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: with vllm_runner(self.MODEL, dtype="half", enforce_eager=eager_mode) as llm: