基于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天|贪心

相关内容

热门资讯

欢乐棋牌!wepower有辅助... 欢乐棋牌!wepower有辅助器(透视挂)苹果版本教程-都是真的有挂(百度贴吧)1、许多玩家不知道欢...
WPk!wepoke透明真的(... WPk!wepoke透明真的(透视辅助)插件挂教程-原来真的有挂(微博热搜)1、操作简单，无需注册，...
aapOKER!wepoke辅... aapOKER!wepoke辅助透视教程(透视辅助)输赢教程-好像真的有挂(头条)1、aapOKER...
AaPOKER!wepoke挂... AaPOKER!wepoke挂(辅助挂)智能教程-原来真的有挂(抖音)；1、AaPOKER机器人多个...
wpk有透视辅助！wepoke... wpk有透视辅助！wepoke辅助是真的，AApoker的确真的有挂，详细教程（有挂教程）；该软件可...
AApOKER!wepoke中... AApOKER!wepoke中牌率(透视挂)系统教程-果真真的有挂(微博热搜)1、AApOKER!w...
微扑克辅助挂！wepoke里面... 微扑克辅助挂！wepoke里面有ai，We辅poker助原来真的有挂，揭秘攻略（有挂攻略）；1、让任...
wepoke有挂！来玩app德... wepoke有挂！来玩app德州辅助器，poker master安卓版的确是有挂的，技巧教程（有挂规...
轰趴十三水!wepoke游戏数... 轰趴十三水!wepoke游戏数据有说法(透视挂)开挂教程-果然真的有挂(知乎)轰趴十三水辅助器中分为...
AaPOKER!wopoker... AaPOKER!wopoker游戏辅助器(透视辅助)黑科技教程-果然真的有挂(哔哩哔哩)1）AaPO...