Skip to content

Commit d4f0fa4

Browse files
authored
fix dy2sta (#1724)
1 parent dbd26fe commit d4f0fa4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddlenlp/ops/faster_transformer/sample/plato_export_model_sample.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
from pprint import pprint
2626

2727
from paddlenlp.ops import FasterGPT
28-
from paddlenlp.transformers import UnifiedTransformerModel, UnifiedTransformerLMHeadModel
28+
from paddlenlp.transformers import UnifiedTransformerLMHeadModel, UnifiedTransformerTokenizer
2929
from paddlenlp.ops import FasterUnifiedTransformer
3030

3131
from paddlenlp.utils.log import logger
@@ -139,7 +139,7 @@ def do_predict(args):
139139
args.topp,
140140
args.decoding_strategy,
141141
tokenizer.cls_token_id, # cls/bos
142-
tokenizer.eos_token_id, # eos
142+
tokenizer.sep_token_id, # sep/eos
143143
tokenizer.pad_token_id, # pad
144144
args.num_beams, # num_beams. Used for beam_search.
145145
args.diversity_rate, # diversity rate. Used for beam search.

0 commit comments

Comments
 (0)