Skip to content

Commit

Permalink
small fix (intel-analytics#11978)
Browse files Browse the repository at this point in the history
* fix

* meet comment
  • Loading branch information
rnwang04 authored and cranechu0131 committed Sep 9, 2024
1 parent 2a9401c commit 3c56af2
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 3c56af2

Please sign in to comment.