colosieve
diff --git a/‎llama_cpp/llama.py
Copy file name to clipboardExpand all lines: llama_cpp/llama.py
+2Lines changed: 2 additions & 0 deletions b/‎llama_cpp/llama.py
Copy file name to clipboardExpand all lines: llama_cpp/llama.py
+2Lines changed: 2 additions & 0 deletions
@@ -358,6 +358,7 @@ def _sample(
 
         if logits_processor is not None:
             logits = logits_processor(list(self.eval_tokens), logits)
+            self.eval_logits[-1] = logits
 
         nl_logit = logits[self._token_nl]
         candidates = self._candidates
@@ -855,6 +856,7 @@ def _create_completion(
         if stopping_criteria is not None and stopping_criteria(
             list(self.eval_tokens), self.eval_logits[-1]
         ):
+            text = self.detokenize(completion_tokens)
             finish_reason = "stop"
 
         if self.verbose: