diff --git a/PaddleNLP/seq2seq/variational_seq2seq/model.py b/PaddleNLP/seq2seq/variational_seq2seq/model.py
index 0225ba1d194995c05e24e27a5cfc72420855f960..41ef96fd0480b2fc61315b5610305bb29694420a 100644
--- a/PaddleNLP/seq2seq/variational_seq2seq/model.py
+++ b/PaddleNLP/seq2seq/variational_seq2seq/model.py
@@ -233,7 +233,7 @@ class VAE(object):
             layers.sampling_id(
                 layers.softmax(
                     layers.squeeze(output_layer(x),[1])
-                ),dtype='int'),
+                    ),dtype='int'),
                 depth=self.tar_vocab_size), [1])
 
         if mode == 'train':