Update rejection_sampler.py

ponix-j · web-flow · commit dcc9ae2d7374 · 2025-05-23T10:34:34.000+08:00
diff --git a/vllm_ascend/sample/rejection_sampler.py b/vllm_ascend/sample/rejection_sampler.py
@@ -452,4 +452,5 @@ def sample_recovered_tokens_pytorch(
             if IS_NGRAM:
                 target_probs[token_idx, draft_token_id] = orig_prob
 
+
 rs.expand_batch_to_tokens = expand_batch_to_tokens
