dev

mlc-ai · Apr 26, 2024 · 0542944 · 0542944
1 parent aad13d4
commit 0542944
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mlc_llm/op/top_p_pivot.py b/python/mlc_llm/op/top_p_pivot.py
@@ -305,7 +305,7 @@ def _func(
 
                             pivot[0] = final_pivot[by]
                             lsum[0] = final_lsum[by]
-                            
+
                             for i in T.serial(T.ceildiv(N, BX * TX)):
                                 idx = T.meta_var(i * BX * TX + bx * TX + tx)
                                 if idx < N: