Skip to content

Commit

Permalink
Remove submodule commits
Browse files Browse the repository at this point in the history
  • Loading branch information
shubhambhokare1 committed Feb 5, 2024
1 parent cf69a37 commit c6d489d
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,7 @@ def export_onnx(
)
input_list = inputs.to_list()

# TODO : Investigate whether copy of model if needed
cloned_model = copy.deepcopy(model).to(device)
out = cloned_model(inputs.encoder_input_ids, inputs.decoder_input_ids)
present = out[2]
Expand Down
1 change: 0 additions & 1 deletion onnxruntime/python/tools/transformers/onnx_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -443,7 +443,6 @@ def match_child_path(
node (str): current node name.
child_op_types (str): constraint of child node op_type of each input edge.
child_output_index (list): constraint of input index of each input edge. None means no constraint.
output_name_to_node (dict): dictionary with output name as key, and node as value.
return_indice (list): a list to append the input index
When there is no constraint on input index of an edge.
Expand Down

0 comments on commit c6d489d

Please sign in to comment.