File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed
src/c++/perf_analyzer/genai-perf Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -141,7 +141,7 @@ def _set_artifact_paths(args: argparse.Namespace) -> argparse.Namespace:
141141 if args .artifact_dir == Path (DEFAULT_ARTIFACT_DIR ):
142142 # Preprocess Huggingface model names that include '/' in their model name.
143143 if (args .model is not None ) and ("/" in args .model ):
144- filtered_name = args .model .split ("/" )[ - 1 ]
144+ filtered_name = "_" . join ( args .model .split ("/" ))
145145 logger .info (
146146 f"Model name '{ args .model } ' cannot be used to create artifact "
147147 f"directory. Instead, '{ filtered_name } ' will be used."
Original file line number Diff line number Diff line change @@ -251,19 +251,19 @@ def test_default_profile_export_filepath(
251251 "arg, expected_path" ,
252252 [
253253 (
254- ["--model" , "company /test_model" ],
255- "artifacts/test_model -triton-tensorrtllm-concurrency1" ,
254+ ["--model" , "strange /test_model" ],
255+ "artifacts/strange_test_model -triton-tensorrtllm-concurrency1" ,
256256 ),
257257 (
258258 [
259259 "--model" ,
260- "company /test_model" ,
260+ "hello/world /test_model" ,
261261 "--service-kind" ,
262262 "openai" ,
263263 "--endpoint-type" ,
264264 "chat" ,
265265 ],
266- "artifacts/test_model -openai-chat-concurrency1" ,
266+ "artifacts/hello_world_test_model -openai-chat-concurrency1" ,
267267 ),
268268 ],
269269 )
You can’t perform that action at this time.
0 commit comments