doc

Author: xadupre

_doc/examples/plot_export_tiny_llm_dim01.py

@@ -72,10 +72,14 @@
 # +++++++++++++++++++
 #
 # We try to export with the following options:
+#
 # - cache registration: register cache serialization with
 #   :func:`onnx_diagnostic.torch_export_patches.register_additional_serialization_functions`
+#
 # - oblivious: an option to remove some the exception raises by the exporter
+#
 # - rt: see ``prefer_deferred_runtime_asserts_over_guards`` in :func:`torch.export.export`
+#
 # - cache_patch: patches the model before exporting with
 #   :func:`onnx_diagnostic.torch_export_patches.torch_export_patches`
 #

_doc/examples/plot_export_tiny_llm_dim01_onnx.py

@@ -67,10 +67,14 @@
 # +++++++++++++++++++
 #
 # We try to export with the following options:
+#
 # - cache registration: register cache serialization with
 #   :func:`onnx_diagnostic.torch_export_patches.register_additional_serialization_functions`
+#
 # - oblivious: an option to remove some the exception raises by the exporter
+#
 # - rt: see ``prefer_deferred_runtime_asserts_over_guards`` in :func:`torch.export.export`
+#
 # - cache_patch: patches the model before exporting with
 #   :func:`onnx_diagnostic.torch_export_patches.torch_export_patches`
 #

_doc/examples/plot_export_tiny_llm_dim01_onnx_custom.py

@@ -66,10 +66,14 @@
 # +++++++++++++++++++
 #
 # We try to export with the following options:
+#
 # - cache registration: register cache serialization with
 #   :func:`onnx_diagnostic.torch_export_patches.register_additional_serialization_functions`
+#
 # - oblivious: an option to remove some the exception raises by the exporter
+#
 # - rt: see ``prefer_deferred_runtime_asserts_over_guards`` in :func:`torch.export.export`
+#
 # - cache_patch: patches the model before exporting with
 #   :func:`onnx_diagnostic.torch_export_patches.torch_export_patches`
 #

_unittests/ut_torch_export_patches/test_patch_torch.py

@@ -345,7 +345,7 @@ def forward(self, x, ind1, ind2):
             )
         self.assertEqualArray(expected, ep.module()(*inputs), atol=1e-2)
 
-    @requires_torch("2.7.9999")
+    @requires_torch("2.8.9999")
     @requires_transformers("4.49.9999")
     def test_export_with_patch_tiny_llm_dim_meta(self):
         data = get_untrained_model_with_inputs("arnir0/Tiny-LLM", verbose=0)