File tree Expand file tree Collapse file tree 1 file changed +0
-2
lines changed
tests/integration/defs/perf Expand file tree Collapse file tree 1 file changed +0
-2
lines changed Original file line number Diff line number Diff line change @@ -1646,7 +1646,6 @@ def get_trtllm_bench_command(self, engine_dir):
16461646 benchmark_cmd += [f"--streaming" ]
16471647 #use default yaml config
16481648 if self ._config .backend == "pytorch" :
1649- import yaml
16501649 pytorch_config_path = os .path .join (engine_dir ,
16511650 "extra-llm-api-config.yml" )
16521651 if not os .path .exists (pytorch_config_path ):
@@ -1658,7 +1657,6 @@ def get_trtllm_bench_command(self, engine_dir):
16581657 yaml .dump (config , f , default_flow_style = False )
16591658 benchmark_cmd += [f"--extra_llm_api_options={ pytorch_config_path } " ]
16601659 elif self ._config .backend == "_autodeploy" :
1661- import yaml
16621660 autodeploy_config_path = os .path .join (engine_dir ,
16631661 "extra_llm_api_options.yaml" )
16641662 if not os .path .exists (autodeploy_config_path ):
You can’t perform that action at this time.
0 commit comments