@@ -183,7 +183,7 @@ def run_text_generation_benchmark(model_path, framework, device, args, num_iters
183
183
184
184
# if num_iters == 0, just output warm-up data
185
185
proc_id = os .getpid ()
186
- if args ['interleave ' ] is True :
186
+ if args ['subsequent ' ] is False :
187
187
for num in range (num_iters + 1 ):
188
188
for prompt_idx , input_text in enumerate (input_text_list ):
189
189
if num == 0 :
@@ -277,7 +277,7 @@ def run_image_generation_benchmark(model_path, framework, device, args, num_iter
277
277
278
278
# if num_iters == 0, just output warm-up data
279
279
proc_id = os .getpid ()
280
- if args ['interleave ' ] is True :
280
+ if args ['subsequent ' ] is False :
281
281
for num in range (num_iters + 1 ):
282
282
for image_id , image_param in enumerate (input_image_list ):
283
283
run_image_generation (image_param , num , image_id , pipe , args , iter_data_list , proc_id )
@@ -482,6 +482,12 @@ def get_argprser():
482
482
help = 'if the value is True, input prompts are processed in interleave manner'
483
483
'if the value is False (default), input prompts are processed in subsequent manner'
484
484
)
485
+ parser .add_argument (
486
+ '--subsequent' ,
487
+ action = 'store_true' ,
488
+ help = 'if the value is True, input prompts are processed in subsequent manner'
489
+ 'if the value is False (default), input prompts are processed in interleave manner'
490
+ )
485
491
parser .add_argument ('-od' , '--output_dir' , help = 'Save the input text and generated text, images to files' )
486
492
utils .model_utils .add_stateful_model_arguments (parser )
487
493
0 commit comments