From cf1382392699f0cff80b87d918a9a55ec136f4d9 Mon Sep 17 00:00:00 2001 From: rachguo Date: Thu, 26 Oct 2023 16:17:41 -0700 Subject: [PATCH] remove print msg --- onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py b/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py index 50bba0110..7e0364833 100644 --- a/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py +++ b/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py @@ -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"]) @@ -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)