Downloads · 30 days
0
nanoGPT2-124m
Razvanip/nanoGPT2-124m
nanoGPT2-124m is a machine learning model from Razvanip. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
The checkpoint for my final implementation of nanoGPT from Andrej Karpathy. This model was trained on an RTX4090 for 48 hours.
Access
Public
Updated Mar 24, 2025
Repo size
548 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth548 MB · 100%
At a glance
- License
- apache-2.0
- Access
- Public
- Created
- Mar 19, 2025
- Updated
- Mar 24, 2025
- SHA
- 8fe0bb9e
Model card
From the Hugging Face model README
The checkpoint for my final implementation of nanoGPT from Andrej Karpathy. This model was trained on an RTX4090 for 48 hours.
For more details, check out my GitHub repository
Usage:
import torch
from transformers import GPT2Tokenizer
from gpt2 import GPT,GPTConfig
tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
model=GPT.from_pretrained('nanoGPT2-124m.pth')
model=model.to(device)
prompt = "Hello, I'm a mathematics teacher"
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
generated_ids=model.generate(input_ids.to(device))
print(tokenizer.decode(generated_ids))
- License
- apache-2.0
- Created
- Mar 19, 2025
- Updated
- Mar 24, 2025