diff --git a/src/petals/models/llama/model.py b/src/petals/models/llama/model.py index 5bf68eeb..1971a290 100644 --- 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: