Fix broken example: examples/offline_inference/profiling at scheduler_config (#18117)

This commit is contained in:
Ecthlion_zyy 2025-05-14 14:19:14 +08:00 committed by GitHub
parent 4f8b373225
commit 33011318c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -193,7 +193,7 @@ def run_profile(context: ProfileContext, csv_output: Optional[str],
batch_size = context.batch_size
prompt_len = context.prompt_len
scheduler_config = llm.llm_engine.scheduler_config
scheduler_config = llm.llm_engine.vllm_config.scheduler_config
max_model_len = llm.llm_engine.model_config.max_model_len
max_num_batched_tokens = scheduler_config.max_num_batched_tokens
max_num_seqs = scheduler_config.max_num_seqs