Skip to content

Commit 71c657e

Browse files
committed
revert irrelevant
Signed-off-by: h-guo18 <[email protected]>
1 parent 94cbb2a commit 71c657e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modelopt/torch/speculative/eagle/eagle_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ def modify(
4545
self.eagle_report_acc = eagle_report_acc
4646
self.eagle_reuse_base_decoder = eagle_reuse_base_decoder
4747
self.eagle_loss_decay_factor = eagle_loss_decay_factor
48+
4849
if eagle_architecture_config.get("parallel_draft_step", 1) > 1:
4950
for i in range(eagle_architecture_config.get("parallel_draft_step") - 1):
5051
self.register_buffer(f"mask_token_{i}", torch.tensor(-1))

0 commit comments

Comments
 (0)