supports one more scenario

Author: xadupre

_unittests/ut_torch_models/test_hghub_model.py

@@ -74,6 +74,7 @@ def test_get_untrained_model_with_inputs_phi_2(self):
         )
 
     @hide_stdout()
+    @ignore_errors(OSError)  # connectitivies issues
     def test_get_untrained_model_with_inputs_beit(self):
         mid = "hf-internal-testing/tiny-random-BeitForImageClassification"
         data = get_untrained_model_with_inputs(mid, verbose=1)

onnx_diagnostic/_command_lines_parser.py

@@ -264,6 +264,11 @@ def get_parser_validate() -> ArgumentParser:
         action=BooleanOptionalAction,
         help="applies patches before exporting",
     )
+    parser.add_argument(
+        "--stop-if-static",
+        default=0,
+        help="raises an exception if a dynamic dimension becomes static",
+    )
     parser.add_argument(
         "--trained",
         default=False,
@@ -319,6 +324,7 @@ def _cmd_validate(argv: List[Any]):
             dtype=args.dtype,
             device=args.device,
             patch=args.patch,
+            stop_if_static=args.stop_if_static,
             optimization=args.opt,
             exporter=args.export,
             dump_folder=args.dump_folder,

onnx_diagnostic/torch_models/test_helper.py

@@ -202,6 +202,7 @@ def validate_model(
     optimization: Optional[str] = None,
     quiet: bool = False,
     patch: bool = False,
+    stop_if_static: bool = True,
     dump_folder: Optional[str] = None,
     drop_inputs: Optional[List[str]] = None,
 ) -> Tuple[Dict[str, Union[int, float, str]], Dict[str, Any]]:
@@ -223,7 +224,10 @@ def validate_model(
     :param optimization: optimization to apply to the exported model,
         depend on the the exporter
     :param quiet: if quiet, catches exception if any issue
-    :param patch: applies patches before exporting
+    :param patch: applies patches (``patch_transformers=True``) before exporting,
+        see :func:`onnx_diagnostic.torch_export_patches.bypass_export_some_errors`
+    :param stop_if_static: stops if a dynamic dimension becomes static,
+        see :func:`onnx_diagnostic.torch_export_patches.bypass_export_some_errors`
     :param dump_folder: dumps everything in a subfolder of this one
     :param drop_inputs: drops this list of inputs (given their names)
     :return: two dictionaries, one with some metrics,
@@ -354,7 +358,9 @@ def validate_model(
             if verbose:
                 print("[validate_model] applies patches before exporting")
             with bypass_export_some_errors(  # type: ignore
-                patch_transformers=True, verbose=max(0, verbose - 1)
+                patch_transformers=True,
+                stop_if_static=stop_if_static,
+                verbose=max(0, verbose - 1),
             ) as modificator:
                 data["inputs_export"] = modificator(data["inputs"])  # type: ignore