debug

microsoft · Oct 26, 2023 · ac78716 · ac78716
1 parent 2b7fe88
commit ac78716
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py b/onnxruntime_extensions/tools/pre_post_processing/steps/nlp.py
@@ -199,6 +199,7 @@ def __init__(self, tokenizer_param: TokenizerParam, name: Optional[str] = None,
         """
         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"])