[side-effect] bring back "--cap" argument in chat cli (#1859)

* [side-effect] bring back argument in chat cli * fix * fix
InternLM · Jun 26, 2024 · b23ba4b · b23ba4b
1 parent 052f550
commit b23ba4b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lmdeploy/cli/chat.py b/lmdeploy/cli/chat.py
@@ -75,6 +75,8 @@ def add_parser_turbomind():
         ArgumentHelper.cache_block_seq_len(engine_group)
         ArgumentHelper.rope_scaling_factor(engine_group)
         ArgumentHelper.session_len(engine_group)
+        # other arguments
+        ArgumentHelper.cap(parser)
         # model args
         ArgumentHelper.revision(engine_group)
         ArgumentHelper.download_dir(engine_group)