Views
No views yet
grid/ - Grid tokenization model (PyTorch, HuggingFace format)
config.json, model.safetensors, generation_config.jsongrid/onnx/model.onnx - ONNX export for browser inference (27MB)[BOS, start_lat, start_lng, end_lat, end_lng, SEP, wp1_lat, wp1_lng, wp2_lat, wp2_lng, ..., EOS]1from transformers import GPT2LMHeadModel, GPT2Config
2import torch
3
4config = GPT2Config.from_pretrained("flipbitsnotburgers/routegpt-copenhagen", subfolder="grid")
5model = GPT2LMHeadModel.from_pretrained("flipbitsnotburgers/routegpt-copenhagen", subfolder="grid")
6
7# Encode start/end coordinates
8GRID_SIZE = 2048
9LAT_MIN, LAT_MAX = 55.60, 55.75
10LNG_MIN, LNG_MAX = 12.45, 12.70
11BOS, EOS, SEP = 4096, 4097, 4098
12
13def encode(lat, lng):
14 lat_bin = int((lat - LAT_MIN) / (LAT_MAX - LAT_MIN) * (GRID_SIZE - 1))
15 lng_bin = int((lng - LNG_MIN) / (LNG_MAX - LNG_MIN) * (GRID_SIZE - 1)) + GRID_SIZE
16 return lat_bin, lng_bin
17
18s_lat, s_lng = encode(55.67, 12.57)
19e_lat, e_lng = encode(55.68, 12.58)
20
21prompt = torch.tensor([[BOS, s_lat, s_lng, e_lat, e_lng, SEP]])
22output = model.generate(prompt, max_new_tokens=300, temperature=0.6, top_k=20, eos_token_id=EOS)