diff --git a/benchmark-scripts/benchmark.py b/benchmark-scripts/benchmark.py index 5942c05..2ede4a7 100644 --- a/benchmark-scripts/benchmark.py +++ b/benchmark-scripts/benchmark.py @@ -196,6 +196,7 @@ def main(): env_vars["DEVICE"] = my_args.target_device retail_use_case_root = os.path.abspath(my_args.retail_use_case_root) env_vars["RETAIL_USE_CASE_ROOT"] = retail_use_case_root + env_vars["VLM_WORKLOAD_ENABLED"] = str(os.getenv("LP_VLM_WORKLOAD_ENABLED")) if my_args.density_increment: env_vars["PIPELINE_INC"] = str(my_args.density_increment) if len(target_fps_list) > 1 and container_names_list: @@ -282,4 +283,4 @@ def main(): (parser_string, my_args.parser_script, traceback.format_exc())) if __name__ == '__main__': - main() \ No newline at end of file + main()