File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
paddlespeech/s2t/models/whisper Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -835,8 +835,14 @@ def update(self,
835
835
logprob , token = paddle .topk (
836
836
logprobs [idx ], k = self .beam_size + 1 )
837
837
for logprob , token in zip (logprob , token ):
838
- new_logprob = (sum_logprobs [idx ] + logprob ).tolist ()[0 ]
839
- sequence = tuple (prefix + [token .tolist ()[0 ]])
838
+ # after Paddle 3.0, tolist in 0-D tensor will return a float/int value instead of a list
839
+ new_logprob = (sum_logprobs [idx ] + logprob ).tolist ()
840
+ new_logprob = new_logprob if isinstance (
841
+ new_logprob , float ) else new_logprob [0 ]
842
+ new_token = token .tolist ()
843
+ new_token = new_token if isinstance (new_token ,
844
+ int ) else new_token [0 ]
845
+ sequence = tuple (prefix + [new_token ])
840
846
scores [sequence ] = new_logprob
841
847
sources [sequence ] = idx
842
848
You can’t perform that action at this time.
0 commit comments