File tree Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -350,6 +350,11 @@ def get_model_config_measurements(self, model_config_name):
350350 #
351351 model_name = model_config_name .rsplit ('_' , 2 )[0 ]
352352
353+ # Remote mode has model_name == model_config_name
354+ #
355+ if model_name not in results :
356+ model_name = model_config_name
357+
353358 if model_name not in results or model_config_name not in results [
354359 model_name ]:
355360 raise TritonModelAnalyzerException (
Original file line number Diff line number Diff line change @@ -41,7 +41,7 @@ MODEL_ANALYZER_PROFILE_BASE_ARGS="$MODEL_ANALYZER_PROFILE_BASE_ARGS --triton-htt
4141MODEL_ANALYZER_PROFILE_BASE_ARGS=" $MODEL_ANALYZER_PROFILE_BASE_ARGS --triton-metrics-url http://localhost:${PORTS[2]} /metrics --checkpoint-directory=$CHECKPOINT_DIRECTORY "
4242MODEL_ANALYZER_PROFILE_BASE_ARGS=" $MODEL_ANALYZER_PROFILE_BASE_ARGS --output-model-repository-path $OUTPUT_MODEL_REPOSITORY --override-output-model-repository"
4343
44- MODEL_ANALYZER_ANALYZE_BASE_ARGS=" --analysis-models $MODEL_NAMES - e $EXPORT_PATH --filename-server-only=$FILENAME_SERVER_ONLY --checkpoint-directory=$CHECKPOINT_DIRECTORY "
44+ MODEL_ANALYZER_ANALYZE_BASE_ARGS=" -e $EXPORT_PATH --filename-server-only=$FILENAME_SERVER_ONLY --checkpoint-directory=$CHECKPOINT_DIRECTORY "
4545MODEL_ANALYZER_ANALYZE_BASE_ARGS=" $MODEL_ANALYZER_ANALYZE_BASE_ARGS --filename-model-inference=$FILENAME_INFERENCE_MODEL --filename-model-gpu=$FILENAME_GPU_MODEL "
4646
4747python3 config_generator.py
You can’t perform that action at this time.
0 commit comments