Code example
Text Generator (ONNX Runtime GenAI)
An on-device evaluation harness for Arm-optimized generative ONNX models.
Evaluation Conditions
Text Generation · AWS Graviton G4 · INT4 · HellaSwag (validation)
| Metric | Resultsvs baseline | Change | |
|---|---|---|---|
| End to end latency p50 | 1307.46ms4376.03ms | −3068.57 ms | |
| End to end latency p90 | 1318.86ms4408.70ms | −3089.84 ms | |
| End to end latency p99 | 1338.23ms4453.49ms | −3115.26 ms | |
| Peak memory | 864.55MB4080.35MB | −3215.80 MB | |
| Average memory | 854.32MB4058.90MB | −3204.58 MB | |
| Time to first inference | 1312.96ms4364.16ms | −3051.20 ms | |
| Tokens per second | 66.88tok/s17.71tok/s | +49.17 tok/s | |
| Time to first token | 407.29ms980.35ms | −573.06 ms | |
| HellaSwag (0-shot) | 59.00%59.80% | −0.80% |
The Optimized Model is provided as a reference implementation to demonstrate and evaluate execution and performance on Arm-based systems. It is not a production-ready or supported solution.
Arm’s publication of the Optimized Model does not constitute an endorsement or certification of the Original Model or a representation that the Optimized Model is suitable for production use or any particular purpose.
To the fullest extent permitted by applicable law (i) the Optimized Model is provided “as is.” Arm makes no representations or warranties that the Original Model, the Optimized Model or their outputs are accurate, safe, secure, non-infringing, legally compliant, suitable for production use or fit for any particular purpose; and (ii) Arm will not be liable for any loss or damage arising from or in connection with the Optimized Model, its use or its outputs.
You are responsible for independently evaluating the Optimized Model, its outputs and its suitability for your intended use, including compliance with applicable legal, regulatory, safety and security requirements.
Arm does not commit to provide ongoing support, maintenance or updates for the Optimized Model. Any use of or reliance on the Optimized Model or its outputs is at your own risk.
Use of the Portal is subject to the Arm Website Terms and Conditions of Use.
| Configuration | Baseline | Arm Optimized |
|---|---|---|
| Dataset | HellaSwag (validation) | HellaSwag (validation) |
| Sample count | 8,000 | 8,000 |
| Batch size | 1 | 1 |
| Prompt length | 138 | 138 |
| Generation length | 60 | 60 |
| Runs | 50 | 50 |
| Warmup runs | 10 | 10 |