We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5c55755 commit aa8b0f8Copy full SHA for aa8b0f8
onnx_diagnostic/torch_models/validate.py
@@ -521,6 +521,11 @@ def validate_model(
521
f.write(f"model_id: {model_id}\n------\n")
522
f.write(pprint.pformat(dump_info))
523
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
529
data["input_options"] = iop
530
data["model_options"] = mop
531
data["model_dump_folder"] = dump_folder
0 commit comments