Mac 部署 GPT-2 预训练模型 gpt2-chinese-cluecorpussmall

news/2024/10/25 14:22:34/

文章目录

  • 下载 GPT-2 模型
  • 快速开始 GPT-2

下载 GPT-2 模型

  • https://huggingface.co/uer/gpt2-chinese-cluecorpussmall

在这里插入图片描述

git clone https://huggingface.co/uer/gpt2-chinese-cluecorpussmall
# 或单独下载 LFS
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/uer/gpt2-chinese-cluecorpussmall

快速开始 GPT-2

  • 安装依赖
pip3 install torch torchvision torchaudio
pip3 install transformers
  • python脚本
from transformers import BertTokenizer, GPT2LMHeadModel, TextGenerationPipeline
tokenizer = BertTokenizer.from_pretrained("./gpt2-chinese-cluecorpussmall")
model = GPT2LMHeadModel.from_pretrained("./gpt2-chinese-cluecorpussmall")
text_generator = TextGenerationPipeline(model, tokenizer)   
result = text_generator("这是很久之前的事情了", max_length=100, do_sample=True)
print(result)
  • 输出结果

在这里插入图片描述


http://www.ppmy.cn/news/1429555.html

相关文章

(十一)PostgreSQL的wal日志(2)-默认wal日志大小

PostgreSQL的wal日志(2)-默认wal日志大小 PostgreSQL的WAL(Write-Ahead Logging)日志文件默认大小为16MB是基于对性能和可靠性权衡的结果。这个默认值是在PostgreSQL早期版本中设定的,目的是在维持良好的系统性能和提高数据恢复能力之间找到…

【java毕业设计】 基于Spring Boot+mysql的房屋租赁系统设计与实现(程序源码)-房屋租赁系统

基于Spring Bootmysql的房屋租赁系统设计与实现(程序源码毕业论文) 大家好,今天给大家介绍基于Spring Bootmysql的房屋租赁系统设计与实现,本论文只截取部分文章重点,文章末尾附有本毕业设计完整源码及论文的获取方式。…

数学建模完整版

模型与适用题型 微分方程传染病预测模型 神经网络 层次分析法 粒子群算法 matlab 优劣解距离法

FastJson2中FastJsonHttpMessageConverter找不到类问题

问题描述 如果你最近也在升级FastJson到FastJson2版本,而跟我一样也遇到了FastJsonHttpMessageConverter找不到类问题以及FastJsonConfig找不到问题,那么恭喜你,看完本文,安装完fastjson2、fastjson2-extension、fastjson2-exte…

【软考】UML中的图之类图

目录 1. 说明2. 图示3. 类图使用方式3.1 对系统的词汇建模3.2 对简单的协作建模3.3 对逻辑数据库模式建模 1. 说明 1.类图(Class Diagram)展现了一组对象、接口、协作和它们之间的关系。2.在面向对象系统的建模中所建立的最常见的图是类图。3.类图给出系…

React - 高级用法

React高级用法 Hooks Reducer useReducer: import React, { useReducer, useState } from react;const initialState { count: 0 };function reducer(state, action) {switch (action.type) {case increment:return { count: state.count 1 }case decrement:re…

LabVIEW多设备控制与数据采集系统

LabVIEW多设备控制与数据采集系统 随着科技的进步,自动化测试与控制系统在工业、科研等领域的应用越来越广泛。开发了一种基于LabVIEW平台开发的多设备控制与数据采集系统,旨在解决多设备手动设置复杂、多路数据显示不直观、数据存储不便等问题。通过RS…

错过100w!

非纪实小说。大家就当看个热闹! 他叫阿来,江苏人,西电研究生毕业,跟我来了同一家公司。 研究生薪资6000,本科生4200,他负责游戏开发,主要是做三方游戏框架移植。 平时只是打个照面,寒…