Tinystories 1M

🚀 モデル名（原文未提供，可根据实际补充）

TinyStoriesデータセットで訓練されたモデルで、詳細はhttps://arxiv.org/abs/2305.07759 を参照してください。

🚀 クイックスタート

このモデルは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"

# 入力IDのエンコード
input_ids = tokenizer.encode(prompt, return_tensors="pt")

# 文章の生成
output = model.generate(input_ids, max_length = 1000, num_beams=1)

# 生成された文章のデコード
output_text = tokenizer.decode(output[0], skip_special_tokens=True)

# 生成された文章の表示
print(output_text)