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 of torch.export.Dim.DYNAMIC when returning the dynamic shapes for a specific models, it is a valid definition for torch.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 function convert_dynamic_axes_into_dynamic_shapes to convert dynamic axes into dynamic shapes, add support for T5ForConditionalGeneration

  • #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

  • #11: adds ModelInputs to guess dynamic shapes

  • #9: adds OnnxruntimeEvaluator

  • #8: adds ExtendedReferenceEvaluator

  • #7: improves function investigate_onnxruntime_issue

0.1.0

first version