Skip to content

Commit

Permalink
fix dynamo export bug
Browse files Browse the repository at this point in the history
  • Loading branch information
inisis committed Sep 15, 2024
1 parent 6431174 commit 6756d19
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ def export_value_info_proto(tensor: Tensor, do_type_check: bool) -> onnx.ValueIn
return onnx_tensor

@staticmethod
def export_attributes(attrs: dict, subgraph_tensor_map) -> List[onnx.AttributeProto]:
def export_attributes(attrs: dict, subgraph_tensor_map=None) -> List[onnx.AttributeProto]:
"""Convert function attributes to ONNX AttributeProtos for model export."""
onnx_attrs: List[onnx.AttributeProto] = []
for key, val in attrs.items():
Expand Down Expand Up @@ -192,7 +192,7 @@ def export_attributes(attrs: dict, subgraph_tensor_map) -> List[onnx.AttributePr
return onnx_attrs

@staticmethod
def export_node(node: Node, subgraph_tensor_map) -> onnx.NodeProto:
def export_node(node: Node, subgraph_tensor_map=None) -> onnx.NodeProto:
# Cannot pass in attrs directly as make_node will change the order
"""Static method to convert an internal node to an ONNX node representation."""
onnx_node = onnx.helper.make_node(
Expand Down

0 comments on commit 6756d19

Please sign in to comment.