Просмотр исходного кода

yapf considers this space to be CRITICAL

50h100a 1 год назад
Родитель
Сommit
2e3318c1fa
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      aphrodite/modeling/layers/sampler.py

+ 1 - 1
aphrodite/modeling/layers/sampler.py

@@ -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.