Fix grammar abort & Minor style fixes (#7204)

2025-06-14 22:49:41 -07:00
parent a6305c7d50
commit 38af4f68a9
8 changed files with 19 additions and 24 deletions
@@ -86,8 +86,8 @@ class EAGLEDraftExtendCudaGraphRunner:

            self.seq_lens = torch.ones((self.max_bs,), dtype=torch.int32)
            self.extend_seq_lens = torch.ones((self.max_bs,), dtype=torch.int32)
-            self.accept_length = (
-                torch.ones((self.max_bs,), dtype=torch.int32) * self.num_tokens_per_bs
+            self.accept_length = torch.full(
+                (self.max_bs,), self.num_tokens_per_bs, dtype=torch.int32
            )

        # Capture