Skip to content

Commit 13038d9

Browse files
Update src/twinkle/sampler/vllm_sampler/vllm_sampler.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent a8450de commit 13038d9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/twinkle/sampler/vllm_sampler/vllm_sampler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def __init__(
141141
# fix: On NPU, monkey_patch_model can trigger Triton compatibility errors and abort sampler init.
142142
# fix: Explicitly skip this patch on NPU and keep it for non-NPU paths only.
143143
# NPU platform may trigger triton errors with monkey_patch_model
144-
if not Platform.get_platform().device_prefix().upper() == 'NPU':
144+
if Platform.get_platform().device_prefix() != 'npu':
145145
self._run_in_loop(self.engine.engine.collective_rpc("monkey_patch_model"))
146146

147147
VLLMLoraWeights()(self)

0 commit comments

Comments
 (0)