因为Transformer接口有改变，跟随修改实现

Signed-off-by: liuzhixi <liudiwu@gmail.com>