model_params.json

{"params": {"vocab_size": 65, "block_size": 128, "n_embd": 256, "n_head": 4, "n_blocks": 4, "dropout": 0.2, "device": "cuda"}}