语言: vi 标签:

• vi
• 越南语
• 文本生成
• gpt3
• 语言模型
• 自然语言处理 数据集:
• 越南语 小部件:
• 文本: 越南是一个拥有 管道标签: 文本生成

越南语GPT-Neo小型模型

首个越南语GPT模型

模型描述

GPT-Neo-vi-small是基于EleutherAI复现的GPT-3架构设计的Transformer模型。

训练数据

GPT-Neo-vi-small在新闻数据集上训练完成，该大规模数据集由新闻网站构建，专门用于训练本模型。

使用方法

每次运行都会生成不同的序列：

from transformers import GPTNeoForCausalLM, GPT2Tokenizer
model = GPTNeoForCausalLM.from_pretrained("NlpHUST/gpt-neo-vi-small")
tokenizer = GPT2Tokenizer.from_pretrained("NlpHUST/gpt-neo-vi-small")
prompt = "辛丑春节刚过，多地出现地价上涨现象。市场活跃，掀起难以置信的抢地热潮，引发部委和地方政府的警示。"
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
gen_tokens = model.generate(input_ids, do_sample=True, temperature=1.0, max_length=1024)
gen_text = tokenizer.batch_decode(gen_tokens)[0]
print(gen_text)

联系方式

有关本项目的个人沟通事宜，请联系阮文雅（nha282@gmail.com）。