Skip to content

Commit f357ec6

Browse files
authored
Docker=-= 兼容性验证 (#4018)
* add docker * fix unit error > Type promotion * fix url * add gpu docker * Update Dockerfile * fix pp3.0 0-d tensor problem * 兼容性验证
1 parent f8dc325 commit f357ec6

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

paddlespeech/s2t/models/whisper/whisper.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -835,8 +835,14 @@ def update(self,
835835
logprob, token = paddle.topk(
836836
logprobs[idx], k=self.beam_size + 1)
837837
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])
840846
scores[sequence] = new_logprob
841847
sources[sequence] = idx
842848

0 commit comments

Comments
 (0)