@@ -258,7 +258,7 @@ def _apply_alphabet_soup(
logits_sort[mask] = -float("inf")
# Apply top-k.
- for i,topk in enumerate(k):
+ for i, topk in enumerate(k):
logits_sort[i, topk:] = -float("inf")
# Re-sort the probabilities.