ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix style (#8300)

Browse source
This commit is contained in:
Yina Chen 2023-06-08 16:52:17 +08:00 committed by GitHub
parent 637b72f2ad
commit 7bfa0fcdf9
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/llm/src/bigdl/llm/ggml/model/generation/utils.py
View file

@ -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