diff --git a/evals/benchmark/stresscli/locust/aistress.py b/evals/benchmark/stresscli/locust/aistress.py index ee1b0edd..0a565844 100644 --- a/evals/benchmark/stresscli/locust/aistress.py +++ b/evals/benchmark/stresscli/locust/aistress.py @@ -224,6 +224,7 @@ def on_locust_init(environment, **_kwargs): os.environ["OPEA_EVAL_SEED"] = environment.parsed_options.seed os.environ["OPEA_EVAL_PROMPTS"] = environment.parsed_options.prompts os.environ["OPEA_EVAL_MAX_OUTPUT_TOKENS"] = str(environment.parsed_options.max_output) + os.environ["LLM_MODEL"] = environment.parsed_options.llm_model try: bench_package = __import__(environment.parsed_options.bench_target) except ImportError: