From 12844ffbf6aafed6139d7c385d5ecd064a1cd5c6 Mon Sep 17 00:00:00 2001
From: dreaming-panda <64370186+dreaming-panda@users.noreply.github.com>
Date: Fri, 22 Mar 2024 12:01:57 -0400
Subject: [PATCH] Update SpecTree.py

---
 Tree/SpecTree.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Tree/SpecTree.py b/Tree/SpecTree.py
index 5739042c..f00e03d2 100644
--- a/Tree/SpecTree.py
+++ b/Tree/SpecTree.py
@@ -149,7 +149,7 @@ def accept_step(self, parent_id :int):
             q = softmax(draft_logits / self.temperature, dim=-1)
             r = self.r[pos + (self.ground_truth_len - 1)]
             
-            if p[token] > r * q[token]:
+            if p[token] >= r * q[token]:
                 return (pos + (self.ground_truth_len - 1), None)
             else:
                 p = self.residual_graph(p, q)
@@ -486,4 +486,4 @@ def verbose(self):
     
     
 
-                
\ No newline at end of file
+