Skip to content

Commit

Permalink
remove print msg
Browse files Browse the repository at this point in the history
  • Loading branch information
rachguo authored and rachguo committed Oct 26, 2023
1 parent 177f1db commit cf13823
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,7 +199,6 @@ def __init__(self, tokenizer_param: TokenizerParam, last_output_optional: bool =
"""
outputs = []
if last_output_optional:
print("enter last_output_optional")
outputs.extend(["input_ids", "attention_mask"])
else:
outputs.extend(["input_ids", "attention_mask", "token_type_ids"])
Expand All @@ -221,8 +220,6 @@ def _create_graph_for_step(self, graph: onnx.GraphProto, onnx_opset: int):
def build_output_declare():
output_base = []
for out in self.output_names:
print("xxxxxxxxxxxxxxxxxx")
print(out)
output_base.append(f"int64[{output_shape_str}] {out}")

return ",".join(output_base)
Expand Down

0 comments on commit cf13823

Please sign in to comment.