From 3c56af2df8b246ca625a5e9473e5e50be892c00a Mon Sep 17 00:00:00 2001 From: Ruonan Wang Date: Fri, 30 Aug 2024 02:55:15 -0700 Subject: [PATCH] small fix (#11978) * fix * meet comment --- .../HF-Transformers-AutoModels/Multimodal/minicpm_v_2_6.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/llm/example/NPU/HF-Transformers-AutoModels/Multimodal/minicpm_v_2_6.py b/python/llm/example/NPU/HF-Transformers-AutoModels/Multimodal/minicpm_v_2_6.py index 8c80231b1dd..c5f6bdd463a 100644 --- a/python/llm/example/NPU/HF-Transformers-AutoModels/Multimodal/minicpm_v_2_6.py +++ b/python/llm/example/NPU/HF-Transformers-AutoModels/Multimodal/minicpm_v_2_6.py @@ -39,8 +39,8 @@ parser.add_argument("--max-output-len", type=int, default=1024) parser.add_argument("--max-prompt-len", type=int, default=960) parser.add_argument("--disable-transpose-value-cache", action="store_true", default=False) - parser.add_argument("--intra-pp", type=int, default=2) - parser.add_argument("--inter-pp", type=int, default=2) + parser.add_argument("--intra-pp", type=int, default=None) + parser.add_argument("--inter-pp", type=int, default=None) args = parser.parse_args() model_path = args.repo_id_or_model_path