diff --git a/CHANGELOGS.rst b/CHANGELOGS.rst index 1d3d28d4..4f717822 100644 --- a/CHANGELOGS.rst +++ b/CHANGELOGS.rst @@ -1,6 +1,9 @@ Change Logs =========== +0.7.15 +++++++ + 0.7.14 ++++++ diff --git a/_doc/index.rst b/_doc/index.rst index 0a8b5df8..e6b595b1 100644 --- a/_doc/index.rst +++ b/_doc/index.rst @@ -239,8 +239,8 @@ The function replaces dynamic dimensions defined as strings by Older versions ============== +* `0.7.15 <../v0.7.15/index.html>`_ * `0.7.14 <../v0.7.14/index.html>`_ -* `0.7.13 <../v0.7.13/index.html>`_ * `0.7.12 <../v0.7.12/index.html>`_ * `0.7.11 <../v0.7.11/index.html>`_ * `0.6.3 <../v0.6.3/index.html>`_ diff --git a/_unittests/ut_tasks/test_tasks.py b/_unittests/ut_tasks/test_tasks.py index 1be4b442..a5300317 100644 --- a/_unittests/ut_tasks/test_tasks.py +++ b/_unittests/ut_tasks/test_tasks.py @@ -39,6 +39,10 @@ def test_text_generation(self): mid = "arnir0/Tiny-LLM" data = get_untrained_model_with_inputs(mid, verbose=1, add_second_input=True) self.assertEqual(data["task"], "text-generation") + self.assertIn("inputs", data) + self.assertIn("inputs2", data) + self.assertIn("inputs_batch1", data) + self.assertIn("inputs_empty_cache", data) self.assertIn((data["size"], data["n_weights"]), [(51955968, 12988992)]) model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"] model(**inputs) diff --git a/onnx_diagnostic/__init__.py b/onnx_diagnostic/__init__.py index 4b0050c2..13c0e5f1 100644 --- a/onnx_diagnostic/__init__.py +++ b/onnx_diagnostic/__init__.py @@ -3,5 +3,5 @@ Functions, classes to dig into a model when this one is right, slow, wrong... """ -__version__ = "0.7.14" +__version__ = "0.7.15" __author__ = "Xavier Dupré" diff --git a/pyproject.toml b/pyproject.toml index 52413972..a7b566cf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "onnx-diagnostic" -version = "0.7.14" +version = "0.7.15" description = "Tools to help converting pytorch models into ONNX." readme = "README.rst" authors = [