From 8f2b4456ad5b7a80be9264fa94927e8a79ed16a9 Mon Sep 17 00:00:00 2001 From: Andrei Betlen Date: Fri, 26 May 2023 22:04:31 -0400 Subject: [PATCH] Format --- llama_cpp/llama.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py index ac51ce5..18372c8 100644 --- a/llama_cpp/llama.py +++ b/llama_cpp/llama.py @@ -217,7 +217,10 @@ class Llama: size = llama_cpp.c_size_t(self._n_vocab) sorted = llama_cpp.c_bool(False) self._candidates_data = np.array( - [], dtype=np.dtype([("id", np.intc), ("logit", np.single), ("p", np.single)], align=True) + [], + dtype=np.dtype( + [("id", np.intc), ("logit", np.single), ("p", np.single)], align=True + ), ) self._candidates_data.resize(3, self._n_vocab) candidates = llama_cpp.llama_token_data_array(