Skip to content

Commit 466c9aa

Browse files
committed
remove copy
1 parent 9cdf2f4 commit 466c9aa

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

llama_cpp/llama.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -989,9 +989,7 @@ def min_length_logits_processor(
989989
scores: npt.NDArray[np.single],
990990
) -> npt.NDArray[np.single]:
991991
if len(input_ids) - len(prompt_tokens) < min_tokens:
992-
new_scores = np.copy(scores)
993-
new_scores[self._token_eos] = -np.inf
994-
return new_scores
992+
scores[self._token_eos] = -np.inf
995993
return scores
996994

997995
_min_length_logits_processor = LogitsProcessorList([min_length_logits_processor])

0 commit comments

Comments
 (0)