mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-14 06:27:45 +08:00
[None][fix] Fix UnboundLocalError. (#8756)
Signed-off-by: Yuxian Qiu <142763828+yuxianq@users.noreply.github.com>
This commit is contained in:
parent
ae57738bae
commit
3176bd3815
@ -1646,7 +1646,6 @@ class MultiMetricPerfTest(AbstractPerfScriptTestClass):
|
||||
benchmark_cmd += [f"--streaming"]
|
||||
#use default yaml config
|
||||
if self._config.backend == "pytorch":
|
||||
import yaml
|
||||
pytorch_config_path = os.path.join(engine_dir,
|
||||
"extra-llm-api-config.yml")
|
||||
if not os.path.exists(pytorch_config_path):
|
||||
@ -1658,7 +1657,6 @@ class MultiMetricPerfTest(AbstractPerfScriptTestClass):
|
||||
yaml.dump(config, f, default_flow_style=False)
|
||||
benchmark_cmd += [f"--extra_llm_api_options={pytorch_config_path}"]
|
||||
elif self._config.backend == "_autodeploy":
|
||||
import yaml
|
||||
autodeploy_config_path = os.path.join(engine_dir,
|
||||
"extra_llm_api_options.yaml")
|
||||
if not os.path.exists(autodeploy_config_path):
|
||||
|
||||
Loading…
Reference in New Issue
Block a user