Downloads · 30 days
32
1% of all-time downloads
rudyon/rudygpt
rudygpt is a text generation model from rudyon. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
A 124M parameter causal language model trained from scratch using rudyon/pipeline on the HuggingFaceFW/fineweb-edu dataset.
Downloads · 30 days
32
1% of all-time downloads
All-time downloads
5.5K
Public
Parameters
137M
3.2 GB on disk
Likes
2
Public
Click a slice to open those files.
.pt1.5 GB · 58%
From the Hugging Face model README
A 124M parameter causal language model trained from scratch using rudyon/pipeline on the HuggingFaceFW/fineweb-edu dataset.
Training was done on a vast.ai instance with 2x 4090S Ti. Training cost was about ~$10.
import torch
from transformers import AutoTokenizer
# download model.py and pytorch_model.bin manually or via hf_hub_download
from model import GPT, GPTConfig
device = 'cuda' if torch.cuda.is_available() else 'cpu'
tokenizer = AutoTokenizer.from_pretrained("rudyon/rudygpt")
model = GPT(GPTConfig(depth=12, vocab_size=50304))
state_dict = torch.load("pytorch_model.bin", map_location=device)
model.load_state_dict(state_dict)
model.eval()
print(model.generate("Hello!"))