Skip to content

Commit aa8b0f8

Browse files
committed
revert unintentional changes
1 parent 5c55755 commit aa8b0f8

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

onnx_diagnostic/torch_models/validate.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -521,6 +521,11 @@ def validate_model(
521521
f.write(f"model_id: {model_id}\n------\n")
522522
f.write(pprint.pformat(dump_info))
523523

524+
# modelbuilder needs different treatments sometimes, so
525+
# we mark it for later usage.
526+
# for example, it has different past_kv ordering than
527+
# flattened CacheObject
528+
data["exporter"] = exporter
524529
data["input_options"] = iop
525530
data["model_options"] = mop
526531
data["model_dump_folder"] = dump_folder

0 commit comments

Comments
 (0)