Downloads · 30 days
208
23% of all-time downloads
Bwenge840/tinystories-gpt-30m
tinystories-gpt-30m is a text generation model from Bwenge840. Use it when you need the model to write or continue text. The card lists the license as mit.
Custom 30M parameter GPT model trained from scratch on the TinyStories dataset.
Downloads · 30 days
208
23% of all-time downloads
All-time downloads
905
Public
Parameters
30.1M
603 MB on disk
Likes
0
Public
Click a slice to open those files.
.pt121 MB · 49%
From the Hugging Face model README
Custom 30M parameter GPT model trained from scratch on the TinyStories dataset.
block_size): 512 subword tokensn_embd): 384tiktoken GPT-2 vocabulary)bfloat16/float16)from transformers import GPT2Tokenizer, pipeline
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
generator = pipeline("text-generation", model="Bwenge840/tinystories-gpt-30m", tokenizer=tokenizer)
story = generator("Once upon a time, a little girl named Jessica", max_new_tokens=150, do_sample=True, top_k=40, temperature=0.8)
print(story[0]["generated_text"])