()
| 97 | dummy_prompts = [{"prompt_token_ids": batch} for batch in dummy_prompt_token_ids.tolist()] |
| 98 | |
| 99 | def llm_generate(): |
| 100 | llm.generate(dummy_prompts, sampling_params=sampling_params, use_tqdm=False, stream=True) |
| 101 | |
| 102 | def run_to_completion(): |
| 103 | start_time = time.perf_counter() |
no test coverage detected