"git@developer.sourcefind.cn:one/TransferBench.git" did not exist on "004710fb7d0492c6fdce50a73def71dcdfebe62f"
Unverified Commit 85d2f662 authored by yunzhongyan0's avatar yunzhongyan0 Committed by GitHub
Browse files

FIX: fix stop_session func bug (#578)



* FIX: fix stop_session func bug

* keep sequence_end = False

---------
Co-authored-by: default avatarhonglei.yan <honglei.yan@nio.com>
Co-authored-by: default avatarAllentDan <AllentDan@yeah.net>
parent c15fbf47
...@@ -268,7 +268,7 @@ class TurboMindInstance: ...@@ -268,7 +268,7 @@ class TurboMindInstance:
self.model_insts[0].register_callback(self._forward_callback) self.model_insts[0].register_callback(self._forward_callback)
if len(input_ids) == 0: if len(input_ids) == 0:
input_ids = [] input_ids = [[]]
if isinstance(input_ids[0], int): if isinstance(input_ids[0], int):
input_ids = [input_ids] input_ids = [input_ids]
...@@ -383,7 +383,7 @@ class TurboMindInstance: ...@@ -383,7 +383,7 @@ class TurboMindInstance:
""" """
if len(input_ids) == 0: if len(input_ids) == 0:
input_ids = [] input_ids = [[]]
if isinstance(input_ids[0], int): if isinstance(input_ids[0], int):
input_ids = [input_ids] input_ids = [input_ids]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment