Views
No views yet
1>>> from modeling_cmpt import BartForConditionalGeneration
2>>> from transformers import AutoTokenizer
3>>> tokenizer = AutoTokenizer.from_pretrained('./CMTP')
4>>> model = BartForConditionalGeneration.from_pretrained('./CMTP')
5>>> inputs = tokenizer.encode("Hello world, 你好 世界", return_tensors='pt')
6>>> pred_ids = model.generate(input_ids, num_beams=4, max_length=20)
7>>> print(tokenizer.convert_ids_to_tokens(pred_ids[i]))
8