From 7bfa0fcdf94a4236a71b8289b5ac0f03f51c183a Mon Sep 17 00:00:00 2001 From: Yina Chen <33650826+cyita@users.noreply.github.com> Date: Thu, 8 Jun 2023 16:52:17 +0800 Subject: [PATCH] fix style (#8300) --- python/llm/src/bigdl/llm/ggml/model/generation/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/llm/src/bigdl/llm/ggml/model/generation/utils.py b/python/llm/src/bigdl/llm/ggml/model/generation/utils.py index e033c7f9..9cd12576 100644 --- a/python/llm/src/bigdl/llm/ggml/model/generation/utils.py +++ b/python/llm/src/bigdl/llm/ggml/model/generation/utils.py @@ -56,7 +56,7 @@ class GenerationMixin: if not is_batched: result = result[0] return result - + def decode(self, tokens: List[int]) -> str: ''' Decode the id to words @@ -74,7 +74,7 @@ class GenerationMixin: return self.detokenize(tokens).decode() def batch_decode(self, - tokens: Union[List[int], List[List[int]]]) -> str: + tokens: Union[List[int], List[List[int]]]) -> str: ''' Decode the id to words