diff --git a/examples/offline_inference/mimo_audio/end2end.py b/examples/offline_inference/mimo_audio/end2end.py index bdcfcd52b..ae044d2e8 100644 --- a/examples/offline_inference/mimo_audio/end2end.py +++ b/examples/offline_inference/mimo_audio/end2end.py @@ -189,7 +189,6 @@ def main(args): batch_timeout=args.batch_timeout, init_timeout=args.init_timeout, shm_threshold_bytes=args.shm_threshold_bytes, - enable_diffusion_pipeline_profiler=args.enable_diffusion_pipeline_profiler, ) thinker_sampling_params = SamplingParams( @@ -434,11 +433,6 @@ def parse_args(): default="../../../model_executor/stage_configs/mimo_audio.yaml", help="Path to a stage configs file.", ) - parser.add_argument( - "--enable-diffusion-pipeline-profiler", - action="store_true", - help="Enable diffusion pipeline profiler to display stage durations.", - ) return parser.parse_args() diff --git a/examples/offline_inference/qwen2_5_omni/end2end.py b/examples/offline_inference/qwen2_5_omni/end2end.py index 9ae99793c..7bba59983 100644 --- a/examples/offline_inference/qwen2_5_omni/end2end.py +++ b/examples/offline_inference/qwen2_5_omni/end2end.py @@ -327,7 +327,6 @@ def main(args): batch_timeout=args.batch_timeout, init_timeout=args.init_timeout, shm_threshold_bytes=args.shm_threshold_bytes, - enable_diffusion_pipeline_profiler=args.enable_diffusion_pipeline_profiler, ) thinker_sampling_params = SamplingParams( temperature=0.0, # Deterministic - no randomness @@ -541,11 +540,6 @@ def parse_args(): default=False, help="Use py_generator mode. The returned type of Omni.generate() is a Python Generator object.", ) - parser.add_argument( - "--enable-diffusion-pipeline-profiler", - action="store_true", - help="Enable diffusion pipeline profiler to display stage durations.", - ) return parser.parse_args() diff --git a/examples/offline_inference/qwen3_omni/end2end.py b/examples/offline_inference/qwen3_omni/end2end.py index d829295e6..155eca4ed 100644 --- a/examples/offline_inference/qwen3_omni/end2end.py +++ b/examples/offline_inference/qwen3_omni/end2end.py @@ -301,7 +301,6 @@ def main(args): log_stats=args.log_stats, stage_init_timeout=args.stage_init_timeout, init_timeout=args.init_timeout, - enable_diffusion_pipeline_profiler=args.enable_diffusion_pipeline_profiler, ) thinker_sampling_params = SamplingParams( @@ -538,11 +537,6 @@ def parse_args(): default=False, help="Use py_generator mode. The returned type of Omni.generate() is a Python Generator object.", ) - parser.add_argument( - "--enable-diffusion-pipeline-profiler", - action="store_true", - help="Enable diffusion pipeline profiler to display stage durations.", - ) parser.add_argument( "--enable-profiler", action="store_true", diff --git a/examples/offline_inference/qwen3_omni/end2end_async_chunk.py b/examples/offline_inference/qwen3_omni/end2end_async_chunk.py index 3eaf30a1c..8adbae9eb 100644 --- a/examples/offline_inference/qwen3_omni/end2end_async_chunk.py +++ b/examples/offline_inference/qwen3_omni/end2end_async_chunk.py @@ -382,7 +382,6 @@ async def run_all(args): stage_configs_path=args.stage_configs_path, log_stats=args.log_stats, stage_init_timeout=args.stage_init_timeout, - enable_diffusion_pipeline_profiler=args.enable_diffusion_pipeline_profiler, ) # Use default sampling params from stage config (they are pre-configured @@ -585,11 +584,6 @@ def parse_args(): default=16000, help="Sampling rate for audio loading.", ) - parser.add_argument( - "--enable-diffusion-pipeline-profiler", - action="store_true", - help="Enable diffusion pipeline profiler to display stage durations.", - ) return parser.parse_args() diff --git a/examples/offline_inference/qwen3_tts/end2end.py b/examples/offline_inference/qwen3_tts/end2end.py index 72ae67b48..901418c39 100644 --- a/examples/offline_inference/qwen3_tts/end2end.py +++ b/examples/offline_inference/qwen3_tts/end2end.py @@ -371,7 +371,6 @@ def main(args): stage_configs_path=args.stage_configs_path, log_stats=args.log_stats, stage_init_timeout=args.stage_init_timeout, - enable_diffusion_pipeline_profiler=args.enable_diffusion_pipeline_profiler, ) batch_size = args.batch_size @@ -393,7 +392,6 @@ async def main_streaming(args): stage_configs_path=args.stage_configs_path, log_stats=args.log_stats, stage_init_timeout=args.stage_init_timeout, - enable_diffusion_pipeline_profiler=args.enable_diffusion_pipeline_profiler, ) for i, prompt in enumerate(inputs): @@ -535,11 +533,6 @@ def parse_args(): default=1, help="Number of prompts per batch (default: 1, sequential).", ) - parser.add_argument( - "--enable-diffusion-pipeline-profiler", - action="store_true", - help="Enable diffusion pipeline profiler to display stage durations.", - ) return parser.parse_args()