File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
paddlenlp/ops/faster_transformer/sample Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 25
25
from pprint import pprint
26
26
27
27
from paddlenlp .ops import FasterGPT
28
- from paddlenlp .transformers import UnifiedTransformerModel , UnifiedTransformerLMHeadModel
28
+ from paddlenlp .transformers import UnifiedTransformerLMHeadModel , UnifiedTransformerTokenizer
29
29
from paddlenlp .ops import FasterUnifiedTransformer
30
30
31
31
from paddlenlp .utils .log import logger
@@ -139,7 +139,7 @@ def do_predict(args):
139
139
args .topp ,
140
140
args .decoding_strategy ,
141
141
tokenizer .cls_token_id , # cls/bos
142
- tokenizer .eos_token_id , # eos
142
+ tokenizer .sep_token_id , # sep/ eos
143
143
tokenizer .pad_token_id , # pad
144
144
args .num_beams , # num_beams. Used for beam_search.
145
145
args .diversity_rate , # diversity rate. Used for beam search.
You can’t perform that action at this time.
0 commit comments