Skip to content

Commit 605dc89

Browse files
authored
Fix npu qwen3moe grpo (#118)
1 parent a6ad6fe commit 605dc89

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/twinkle/sampler/vllm_sampler/vllm_sampler.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,8 +122,7 @@ def __init__(self, model_id: str, engine_args: Dict[str, Any] = None, device_mes
122122
# fix: On NPU, monkey_patch_model can trigger Triton compatibility errors and abort sampler init.
123123
# fix: Explicitly skip this patch on NPU and keep it for non-NPU paths only.
124124
# NPU platform may trigger triton errors with monkey_patch_model
125-
if Platform.get_platform().device_prefix() != 'npu':
126-
self._run_in_loop(self.engine.engine.collective_rpc('monkey_patch_model'))
125+
self._run_in_loop(self.engine.engine.collective_rpc('monkey_patch_model'))
127126

128127
VLLMLoraWeights()(self)
129128

0 commit comments

Comments
 (0)