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)