Skip to content

Commit

Permalink
Force export LLM_Model environment variable
Browse files Browse the repository at this point in the history
Signed-off-by: lvliang-intel <[email protected]>
  • Loading branch information
lvliang-intel committed Oct 24, 2024
1 parent 516b415 commit 302c32e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions evals/benchmark/stresscli/locust/aistress.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 302c32e

Please sign in to comment.