From c54c9506aa7f60839fad2d4d586b2943da46949b Mon Sep 17 00:00:00 2001 From: zxcd <228587199@qq.com> Date: Tue, 13 Dec 2022 21:14:47 +0800 Subject: [PATCH] remove paddle.fluid (#2741) --- paddlespeech/s2t/models/whisper/whipser.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/paddlespeech/s2t/models/whisper/whipser.py b/paddlespeech/s2t/models/whisper/whipser.py index ba998333..63cafbdb 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) -- GitLab