Black formatting

This commit is contained in:
Andrei Betlen
2023-03-24 14:59:29 -04:00
parent c784d83131
commit 7786edb0f9
4 changed files with 17 additions and 3 deletions

View File

@@ -60,7 +60,11 @@ class Llama:
stop = [s.encode("utf-8") for s in stop]
prompt_tokens = llama_cpp.llama_tokenize(
self.ctx, prompt.encode("utf-8"), self.tokens, llama_cpp.llama_n_ctx(self.ctx), True
self.ctx,
prompt.encode("utf-8"),
self.tokens,
llama_cpp.llama_n_ctx(self.ctx),
True,
)
if prompt_tokens + max_tokens > self.params.n_ctx: