Skip to content

Commit

Permalink
Resync dev with main (#65)
Browse files Browse the repository at this point in the history
* 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
EthanMarx and William Benoit authored Oct 3, 2024
1 parent ee891d1 commit ff0fb11
Showing 0 changed files with 0 additions and 0 deletions.

0 comments on commit ff0fb11

Please sign in to comment.