| ONNX Model Export | |
| Generated: 2025-12-28T17:48:01.525Z | |
| Model: micro-distill-grpo-vae | |
| Framework: PyTorch 2.0+ | |
| ONNX Version: 1.14.0 | |
| Graph Structure: | |
| - Inputs: | |
| - input_ids: int64[batch_size, sequence_length] | |
| - attention_mask: int64[batch_size, sequence_length] | |
| - position_ids: int64[batch_size, sequence_length] (optional) | |
| - Outputs: | |
| - logits: float32[batch_size, sequence_length, 50257] | |
| Layers: | |
| - Embedding: 512 dimensions | |
| - 8 Transformer blocks | |
| - Layer normalization | |
| - Language modeling head | |
| Optimizations: | |
| - Constant folding: enabled | |
| - Shape inference: enabled | |
| - Dynamic axes: sequence_length | |
| - Opset: 17 | |
| - IR version: 9 | |
| Quantization: Q4 (4-bit) |