Skip to content

Commit be0add1

Browse files
committed
Fix type bug
1 parent e214a58 commit be0add1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

llama_cpp/llama.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -491,7 +491,7 @@ def set_rng_seed(self, seed: int):
491491
def sample_repetition_penalties(
492492
self,
493493
candidates: "_LlamaTokenDataArray",
494-
last_tokens_data: llama_cpp.Array[llama_cpp.llama_token],
494+
last_tokens_data: "llama_cpp.Array[llama_cpp.llama_token]",
495495
penalty_last_n: int,
496496
penalty_repeat: float,
497497
penalty_freq: float,

0 commit comments

Comments
 (0)