1from model import TinyLLM
2from tokenizer import Tokenizer
3import torch
4
5# Load tokenizer
6tokenizer = Tokenizer.load("tokenizer.json")
7
8# Load model
9checkpoint = torch.load("model.pt", map_location="cpu")
10model = TinyLLM(
11 vocab_size=74,
12 d_model=512,
13 num_heads=8,
14 num_layers=6,
15 d_ff=1024,
16 max_seq_len=512,
17 dropout=0.1,
18 pad_token_id=0
19)
20model.load_state_dict(checkpoint["model_state_dict"])
21model.eval()
22
23# Generate text
24prompt = "What is Mach 0.3"
25input_ids = tokenizer.encode(prompt, add_special_tokens=False)
26input_ids = torch.tensor([input_ids], dtype=torch.long)
27output = model.generate(input_ids, max_new_tokens=150, temperature=0.8)
28print(tokenizer.decode(output[0].tolist()))
Training data: Capt Ajit Vadakayil's writings
Model: Custom transformer implementation