Skip to content

Commit 8f2b445

Browse files
committed
Format
1 parent 84e313b commit 8f2b445

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

llama_cpp/llama.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,10 @@ def __init__(
217217
size = llama_cpp.c_size_t(self._n_vocab)
218218
sorted = llama_cpp.c_bool(False)
219219
self._candidates_data = np.array(
220-
[], dtype=np.dtype([("id", np.intc), ("logit", np.single), ("p", np.single)], align=True)
220+
[],
221+
dtype=np.dtype(
222+
[("id", np.intc), ("logit", np.single), ("p", np.single)], align=True
223+
),
221224
)
222225
self._candidates_data.resize(3, self._n_vocab)
223226
candidates = llama_cpp.llama_token_data_array(

0 commit comments

Comments
 (0)