 genius-chinese from transformers import BertTokenizer, BartForConditionalGeneration, Text2TextGenerationPipeline checkpoint = 'beyond/genius-base-chinese' tokenizer = BertTokenizer.from_pretrained(checkpoint) genius_model = BartForConditionalGeneration.from_pretrained(checkpoint) genius_generator = Text2TextGenerationPipeline(genius_model, tokenizer, device=0) genius_generator  sketchs = [     "今天[MASK]篮球[MASK]学校[MASK]",     "自然语言处理[MASK]谷歌[MASK]通用人工智能[MASK]",     "我昨天做了一个梦[MASK]又遇见了她[MASK]曾经那段时光让人怀恋[MASK]",     "[MASK]疫情[MASK]公园[MASK]散步[MASK]",     "[MASK]酸菜鱼火锅[MASK]很美味，味道绝了[MASK]周末真开心[MASK]"     "" ] for sketch in sketchs:     print('input sketch:\n>>> ', sketch)     print('genius-chinese output:\n>>> ',genius_generator(sketch, max_length=100, do_sample=True, num_beams=3)[0]['generated_text'].replace(' ',''),'\n') ```  