| { | |
| "runtime": "tinybrain-causallm-v1", | |
| "model_id": "sshleifer/tiny-gpt2", | |
| "bundle_name": "tinygpt2_only_logits", | |
| "architecture": "gpt2", | |
| "sequence_length": 16, | |
| "vocab_size": 50257, | |
| "input_names": [ | |
| "input_ids", | |
| "attention_mask" | |
| ], | |
| "output_name": "logits", | |
| "tokenizer_dir": "tokenizer", | |
| "requires_compiled_model": true | |
| } |