Set fallback=False by default in ONNX export

justinchuby · justinchuby · commit 8b9acb6aed5e · 2025-09-11T07:51:26.000-07:00
Co-authored-by: justinchuby &lt;11205048+justinchuby@users.noreply.github.com&gt;
diff --git a/torch/onnx/__init__.py b/torch/onnx/__init__.py
@@ -81,7 +81,7 @@ def export(
     profile: bool = False,
     dump_exported_program: bool = False,
     artifacts_dir: str | os.PathLike = ".",
-    fallback: bool = True,
+    fallback: bool = False,
     # BC options
     export_params: bool = True,
     keep_initializers_as_inputs: bool = False,
diff --git a/torch/onnx/_internal/exporter/_compat.py b/torch/onnx/_internal/exporter/_compat.py
@@ -67,7 +67,7 @@ def export_compat(
     profile: bool = False,
     dump_exported_program: bool = False,
     artifacts_dir: str | os.PathLike = ".",
-    fallback: bool = True,
+    fallback: bool = False,
     # Legacy export parameters for fallback
     legacy_export_kwargs: dict[str, Any] | None = None,
 ) -> _onnx_program.ONNXProgram:
