fix artifact path

Author: nv-hwoo

src/c++/perf_analyzer/genai-perf/genai_perf/parser.py

@@ -139,7 +139,17 @@ def _set_artifact_paths(args: argparse.Namespace) -> argparse.Namespace:
     Set paths for all the artifacts.
     """
     if args.artifact_dir == Path(DEFAULT_ARTIFACT_DIR):
-        name = [f"{args.model}"]
+        # Preprocess Huggingface model names that include '/' in their model name.
+        if "/" in args.model:
+            filtered_name = args.model.split("/")[-1]
+            logger.info(
+                f"Model name '{args.model}' cannot be used to create artifact "
+                f"directory. Instead, '{filtered_name}' will be used."
+            )
+            name = [f"{filtered_name}"]
+        else:
+            name = [f"{args.model}"]
+
         if args.service_kind == "openai":
             name += [f"{args.service_kind}-{args.endpoint_type}"]
         elif args.service_kind == "triton":

src/c++/perf_analyzer/genai-perf/tests/test_cli.py

@@ -247,6 +247,35 @@ def test_default_profile_export_filepath(
         captured = capsys.readouterr()
         assert captured.out == ""
 
+    @pytest.mark.parametrize(
+        "arg, expected_path",
+        [
+            (
+                ["--model", "company/test_model"],
+                "artifacts/test_model-triton-tensorrtllm-concurrency1",
+            ),
+            (
+                [
+                    "--model",
+                    "company/test_model",
+                    "--service-kind",
+                    "openai",
+                    "--endpoint-type",
+                    "chat",
+                ],
+                "artifacts/test_model-openai-chat-concurrency1",
+            ),
+        ],
+    )
+    def test_model_name_artifact_path(self, monkeypatch, arg, expected_path, capsys):
+        combined_args = ["genai-perf"] + arg
+        monkeypatch.setattr("sys.argv", combined_args)
+        args, extra_args = parser.parse_args()
+
+        assert args.artifact_dir == Path(expected_path)
+        captured = capsys.readouterr()
+        assert captured.out == ""
+
     def test_default_load_level(self, monkeypatch, capsys):
         monkeypatch.setattr("sys.argv", ["genai-perf", "--model", "test_model"])
         args, extra_args = parser.parse_args()