Change Logs¶
0.4.0¶
#61: improves dynamic shapes for EncoderDecoderCache
#58: add function use_dyn_not_str to replace string by
torch.export.Dim.DYNAMIC
, use string instead oftorch.export.Dim.DYNAMIC
when returning the dynamic shapes for a specific models, it is a valid definition fortorch.onnx.export
which can reuse the names#55: add support for text-classification
#54: add support for fill-mask, refactoring
#52: add support for zero-shot-image-classification
#50: add support for onnxruntime fusion
#48: add support for EncoderDecoderCache, test with openai/whisper-tiny
#45: improve change_dynamic_dimension to fix some dimensions
0.3.0¶
#43: uses custom patches
#38: uses the registered serialization functions when it is available
#30, #31: adds command to test a model id, validate the export
#29: adds helpers to measure the memory peak and run benchmark on different processes
#28: adds command line to print out the configuration for a model id, support image-text-to-text
#26: creates a folder
helpers
to gather all the functions used in many places#25: improve patches for DynamicCache (issue with register_pytree_flatten_spec being deprecated)
#24: dummy inputs for
text2text-generation
, add new functionconvert_dynamic_axes_into_dynamic_shapes
to convert dynamic axes into dynamic shapes, add support forT5ForConditionalGeneration
#23: dummy inputs for
image-classification
#22, #27: api to create untrained model copying the architecture of the trained models and dummy inputs for them, support for
text-generation
0.2.1¶
#16: refactors patches, add model Phi2, implements a tweak to raise an exception with a dynamic dimension becomes static when exporting a model
0.2.0¶
0.1.0¶
first version