基于MindFormers实现GPT2模型的推理
创始人
2024-11-04 21:08:02
0次

前言

针对MindFormers的安装，可参考本专栏里的另一篇博客

安装MindFormers（昇腾910）-CSDN博客

pipeline方式

from mindformers import pipeline from mindformers import GPT2LMHeadModel, GPT2Config, GPT2Tokenizer tok = GPT2Tokenizer.from_pretrained("gpt2") gpt_model = GPT2LMHeadModel.from_pretrained("gpt2") prompt="I love Shanghai" text_generation = pipeline(task='text_generation', model=gpt_model, tokenizer=tok, max_length=50) outputs=text_generation(prompt, top_k=3) print(outputs)

持续更新中...

上一篇：算法效率的分析

下一篇：代码随想录第28天|贪心

相关内容

热门资讯

第五分钟科普！wepoker有... 第五分钟科普！wepoker有辅助功能吗，aapoker发牌逻辑，规律教程（有挂细节）-哔哩哔哩1、...
第6分钟了解“中至余干马甲怎么... 第6分钟了解“中至余干马甲怎么开挂”详细透视开挂辅助工具-哔哩哔哩；中至余干马甲怎么开挂最新软件透明...
第2小时了解！兴动休闲辅助，拱... 第2小时了解！兴动休闲辅助，拱趴大菠萝开挂方法，玩家教程（有挂助手）1、起透看视 兴动休闲辅助透明视...
第7分钟了解(甘孜大厅)外挂透... 第7分钟了解(甘孜大厅)外挂透明挂辅助器(辅助挂)2025新版教程(2020已更新)(哔哩哔哩)；超...
第3分钟私人局！wepoker... 第3分钟私人局！wepoker安装教程，wepoker养号规律，存在挂教程（有挂助手）-哔哩哔哩1、...
八分钟了解“蛮王辅助器”详细透... 八分钟了解“蛮王辅助器”详细透视开挂辅助工具-哔哩哔哩;AI辅助机器人普及解说快速成为一个“必胜”的...
第2分钟了解(天天互娱)外挂透... 第2分钟了解(天天互娱)外挂透明挂辅助软件(辅助挂)技巧教程(2023已更新)(哔哩哔哩)；天天互娱...
第2小时了解！九江讨赏辅助，w... 第2小时了解！九江讨赏辅助，werplan有挂吗，线上教程（揭秘有挂）1、实时九江讨赏辅助开挂更新：...
6分钟理解！aapoker辅助... 6分钟理解！aapoker辅助插件工具下载，werplan透视挂，系统教程（有挂规律）-哔哩哔哩1、...
六分钟了解(OB棋牌)外挂辅助... 六分钟了解(OB棋牌)外挂辅助黑科技(透视)解密教程(2025已更新)(哔哩哔哩)；OB棋牌最新软件...