Skip to content

Commit

Permalink
Checking for existence of model repo (because analyze doesn't have to…
Browse files Browse the repository at this point in the history
… provide one)
  • Loading branch information
nv-braf committed Aug 11, 2023
1 parent 6fccc79 commit 19daeda
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions model_analyzer/triton/server/server_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,15 +178,13 @@ def _get_local_server_handle(config, gpus, use_model_repository):
triton_config = TritonServerConfig()
triton_config.update_config(config.triton_server_flags)

assert use_model_repository and config.model_repository

triton_config["model-repository"] = (
config.model_repository
if use_model_repository
if use_model_repository and config.model_repository
else config.output_model_repository_path
)

if use_model_repository:
if use_model_repository and config.model_repository:
triton_config["strict-model-config"] = "false"

triton_config["http-port"] = config.triton_http_endpoint.split(":")[-1]
Expand Down

0 comments on commit 19daeda

Please sign in to comment.