PULI GPT 2

🚀 PULI GPT - 2

PULI GPT - 2是一个匈牙利语的GPT - 2模型，可用于文本生成任务。它使用Megatron - DeepSpeed进行训练，在363亿个单词的数据集上训练了500000步。如需更多详情，请查看我们的演示网站。

🚀 快速开始

你可以按照以下步骤使用PULI GPT - 2模型：

from transformers import GPT2Tokenizer, GPT2Model

tokenizer = GPT2Tokenizer.from_pretrained('NYTK/PULI-GPT-2')
model = GPT2Model.from_pretrained('NYTK/PULI-GPT-2')
text = "Replace me by any text you'd like."
encoded_input = tokenizer(text, return_tensors='pt')
output = model(**encoded_input)

高级用法

使用pipeline进行文本生成：

from transformers import pipeline

prompt = "Elmesélek egy történetet a nyelvtechnológiáról."
generator = pipeline(task="text-generation", model="NYTK/PULI-GPT-2")

print(generator(prompt)[0]["generated_text"])

✨ 主要特性

• 匈牙利语GPT - 2模型，适用于匈牙利语的文本生成任务。
• 使用Megatron - DeepSpeed进行训练，可参考其GitHub仓库。
• 训练数据集包含363亿个单词。
• 训练步数达到500000步。

📚 详细文档

限制条件

• 最大序列长度max_seq_length为1024。

引用信息

如果你使用了该模型，请引用以下论文：

@inproceedings {yang-puli,
    title = {Jönnek a nagyok! BERT-Large, GPT-2 és GPT-3 nyelvmodellek magyar nyelvre},
	booktitle = {XIX. Magyar Számítógépes Nyelvészeti Konferencia (MSZNY 2023)},
	year = {2023},
	publisher = {Szegedi Tudományegyetem, Informatikai Intézet},
	address = {Szeged, Hungary},
	author = {Yang, Zijian Győző and Dodé, Réka and Ferenczi, Gergő and Héja, Enikő and Jelencsik-Mátyus, Kinga and Kőrös, Ádám and Laki, László János and Ligeti-Nagy, Noémi and Vadász, Noémi and Váradi, Tamás},
	pages = {247--262}
}

📄 许可证

本模型采用CC - BY - NC - 4.0许可证。

📦 模型信息