Fix a bug in BatchTokenIDOut & Misc style and dependency updates (#7457)

2025-06-23 06:20:39 -07:00
parent 8aa68ed5c4
commit 55e03b10c4
9 changed files with 37 additions and 32 deletions
--- a/sgl-kernel/python/sgl_kernel/sampling.py
+++ b/sgl-kernel/python/sgl_kernel/sampling.py
@@ -1,7 +1,7 @@
 from typing import Optional, Union

 import torch
-from sgl_kernel.utils import _to_tensor_scalar_tuple, get_cuda_stream
+from sgl_kernel.utils import _to_tensor_scalar_tuple


 def _top_k_renorm_probs_internal(