未验证 提交 c54c9506 编写于 作者: Z zxcd 提交者: GitHub

remove paddle.fluid (#2741)

上级 9064b93e
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册