fix

ztxz16 · Jul 25, 2024 · 9d79e50 · 9d79e50
1 parent 637e50d
commit 9d79e50
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/fastllm_pytools/llm.py b/tools/fastllm_pytools/llm.py
@@ -660,6 +660,7 @@ def token_healing(self,
             for l in range(len(common_token)):
                 if (l >= len(cur_token) or cur_token[l] != common_token[l]):
                     common_token = common_token[:l]
+                    break
 
         last_ret = ""
         max_prob = -1e5