saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-07 17:15:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Metrics - llama-2 - Add package name and version to engine of onnxruntime (#19325)

Browse source 
### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
This commit is contained in:
ironman 2024-02-02 07:52:20 +08:00 committed by GitHub
parent 1bdd7d9499
commit 8a2646ce60
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/transformers/models/llama/benchmark_all.py
View file

@ -253,7 +253,7 @@ def save_results(results, filename):
# Save results to csv with standard format
records = []
for _, row in df.iterrows():
if row["Engine"] == "optimum-ort":
if row["Engine"] in ["optimum-ort", "onnxruntime"]:
record = BenchmarkRecord(
row["Model Name"], row["Precision"], "onnxruntime", row["Device"], ort_pkg_name, ort_pkg_version
)