Docker=-= 兼容性验证 (#4018)

Liyulingyue · web-flow · commit f357ec61720b · 2025-03-18T11:43:12.000+08:00
* add docker

* fix unit error &gt; Type promotion

* fix url

* add gpu docker

* Update Dockerfile

* fix pp3.0 0-d tensor problem

* &#20860;&#23481;&#24615;&#39564;&#35777;
diff --git a/paddlespeech/s2t/models/whisper/whisper.py b/paddlespeech/s2t/models/whisper/whisper.py
@@ -835,8 +835,14 @@ def update(self,
                 logprob, token = paddle.topk(
                     logprobs[idx], k=self.beam_size + 1)
                 for logprob, token in zip(logprob, token):
-                    new_logprob = (sum_logprobs[idx] + logprob).tolist()[0]
-                    sequence = tuple(prefix + [token.tolist()[0]])
+                    # after Paddle 3.0, tolist in 0-D tensor will return a float/int value instead of a list
+                    new_logprob = (sum_logprobs[idx] + logprob).tolist()
+                    new_logprob = new_logprob if isinstance(
+                        new_logprob, float) else new_logprob[0]
+                    new_token = token.tolist()
+                    new_token = new_token if isinstance(new_token,
+                                                        int) else new_token[0]
+                    sequence = tuple(prefix + [new_token])
                     scores[sequence] = new_logprob
                     sources[sequence] = idx
 
