[ONNX] Do not run 'deduplicate_initializers' when 'keep_initializers_…

…as_inputs' is True ghstack-source-id: fead9d663757be87f05487c2816f95bf5c78a4eb Pull Request resolved: #96320
pytorch · Aug 1, 2023 · abd4cfb · abd4cfb
1 parent f23d755
commit abd4cfb
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/torch/onnx/utils.py b/torch/onnx/utils.py
@@ -1603,9 +1603,11 @@ def _export(
                     module_typenames_to_export_as_functions,
                     list(params_dict.keys()),
                 )
-            params_dict = _C._jit_pass_onnx_deduplicate_initializers(  # type: ignore[assignment]
-                graph, params_dict, getattr(model, "training", False)  # type: ignore[arg-type]
-            )
+
+            if keep_initializers_as_inputs is not True:
+                params_dict = _C._jit_pass_onnx_deduplicate_initializers(  # type: ignore[assignment]
+                    graph, params_dict, getattr(model, "training", False)  # type: ignore[arg-type]
+                )
             _C._jit_pass_onnx_assign_scoped_names_for_node_and_value(graph)
             if export_params:
                 (