File tree Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Original file line number Diff line number Diff line change 6363"""
6464
6565import copy
66+ import pprint
6667import torch
6768import transformers
69+ from onnx_diagnostic .helpers import string_type
6870from onnx_diagnostic .torch_export_patches .onnx_export_errors import bypass_export_some_errors
6971from onnx_diagnostic .torch_models .llms import get_tiny_llm
7072
7880
7981cloned_inputs = copy .deepcopy (inputs )
8082
83+ # %%
84+ # Let's show this inputs, this was inferred in
85+ # example :ref:`l-plot-tiny-llm-export`.
86+
87+ print (string_type (inputs , with_shape = True ))
88+
89+ # %%
90+ # And the dynamic shapes
91+ pprint .pprint (dynamic_shapes )
92+
93+ # %%
94+ # We are ready to export.
8195
8296with bypass_export_some_errors (patch_transformers = True ) as modificator :
8397 ep = torch .export .export (
You can’t perform that action at this time.
0 commit comments