fix

bigscience-workshop · Jul 23, 2024 · c142727 · c142727
1 parent 140ee9b
commit c142727
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/petals/models/llama/model.py b/src/petals/models/llama/model.py
@@ -221,7 +221,6 @@ def _sample(
                 if first_token is None:
                     first_token = valid_token
 
-                # Проверка правильности токена)
                 if valid_token.item() == speculative_tokens[:, i].item():
                     all_valid_tokens.append(valid_token.unsqueeze(-1))
                 else: