diff --git a/paddlespeech/s2t/models/whisper/whipser.py b/paddlespeech/s2t/models/whisper/whipser.py index ba998333883e567894a1af20147200d2a275bd2f..63cafbdb726050c9b9b5e03b7d7ba869fe4ece73 100644 --- a/paddlespeech/s2t/models/whisper/whipser.py +++ b/paddlespeech/s2t/models/whisper/whipser.py @@ -16,7 +16,6 @@ from typing import Union import numpy as np import paddle -import paddle.fluid as fluid import paddle.nn.functional as F import soundfile import tqdm @@ -231,8 +230,8 @@ class TextDecoder(nn.Layer): ]) self.ln = LayerNorm(n_state) - mask = fluid.layers.fill_constant( - shape=[n_ctx, n_state], value=-np.inf, dtype='float32') + mask = paddle.full( + shape=[n_ctx, n_state], fill_value=-np.inf, dtype='float32') mask = paddle.triu(mask, diagonal=1) self.register_buffer("mask", mask, persistable=False)