Fix CodeQL error

microsoft · Oct 21, 2023 · d14d5bd · d14d5bd
1 parent e69c23b
commit d14d5bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxruntime/python/tools/transformers/models/llama/llama_parity.py b/onnxruntime/python/tools/transformers/models/llama/llama_parity.py
@@ -29,9 +29,9 @@ def get_sequence_lengths(args: argparse.Namespace):
 def get_inputs(args: argparse.Namespace, config: LlamaConfig):
     # Dummy values for parity
     batch_size = 2
+    past_sequence_length, sequence_length, _ = get_sequence_lengths(args)
 
     if args.merged:
-        past_sequence_length, sequence_length, _ = get_sequence_lengths(args)
         inputs = get_merged_sample_with_past_kv_inputs(
             config,
             args.device,
@@ -47,6 +47,7 @@ def get_inputs(args: argparse.Namespace, config: LlamaConfig):
         )
     else:
         inputs = get_sample_inputs(config, args.device, batch_size, sequence_length, return_dict=True)
+
     return inputs