Tinystories 1M

🚀 TinyStories-1M模型

本项目是在TinyStories数据集上训练的模型，可用于文本生成等任务，为相关自然语言处理场景提供支持。

🚀 快速开始

本模型在TinyStories数据集上进行训练，更多关于该数据集的信息可查看论文：链接

💻 使用示例

基础用法

from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig

model = AutoModelForCausalLM.from_pretrained('roneneldan/TinyStories-1M')

tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125M")

prompt = "Once upon a time there was"

input_ids = tokenizer.encode(prompt, return_tensors="pt")

# Generate completion
output = model.generate(input_ids, max_length = 1000, num_beams=1)

# Decode the completion
output_text = tokenizer.decode(output[0], skip_special_tokens=True)

# Print the generated text
print(output_text)