[Qwen] Remove cuda hard-code in qwen3 next (#25243)

Signed-off-by: Icey <1790571317@qq.com>
2025-09-19 20:25:12 +08:00
parent 5089fd749c
commit 838d7116ba
1 changed files with 1 additions and 1 deletions
--- a/vllm/model_executor/models/qwen3_next.py
+++ b/vllm/model_executor/models/qwen3_next.py
@@ -306,7 +306,7 @@ class Qwen3NextGatedDeltaNet(nn.Module, MambaBase):
            eps=self.layer_norm_epsilon,
            group_size=None,
            norm_before_gate=True,
-            device=torch.cuda.current_device(),
+            device=current_platform.current_device(),
            dtype=config.torch_dtype,
        )