Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Allow parsing input tensors from torch `ScriptModules` (#57) * add ability to parse inputs from script module * update torch dep to ^2.0 for consistency with ml4gw * add onnx dep * remove both tests * remove both tests * Batched state updates (#59) * allow for batched state updates * update poetry lock * restrict tf version * revert whitespaces * Allow python 3.12 and explicitly list PyPI as a source for future compatibility * Add python 3.12 to unit tests * Add py312 to tox file * Missed one * Attempt to update tensorflow * Change tf specification * Force usage of keras 2 * Fix formatting * Remove py38 support and install tf-keras * More intelligent and flexible input and output tensor naming for `TorchScript` Exporter (#62) * support direct torchscript export from ScriptModule * add tests of new naming behavior * guard torch import * add jit module to handles for onnx and torchscript exporters * add script module test to exporter utils * bump to version 0.2.0 (#64) --------- Co-authored-by: William Benoit <[email protected]>
- Loading branch information