Skip to content

Commit 19061c6

Browse files
committed
Adding Compute-Context-Length(CCL)
Signed-off-by: Vahid Janfaza <vjanfaza@qti.qualcomm.com>
1 parent a90a4ec commit 19061c6

File tree

1 file changed

+0
-12
lines changed

1 file changed

+0
-12
lines changed

QEfficient/cloud/infer.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -340,18 +340,6 @@ def main(
340340
"--prompt-len", "--prompt_len", default=32, type=int, help="Sequence length for text generation."
341341
)
342342
parser.add_argument("--ctx-len", "--ctx_len", default=128, type=int, help="Context length for text generation.")
343-
parser.add_argument('--comp-ctx-lengths-prefill',
344-
type=lambda comp_ctx_lengths_prefill:
345-
[int(x) for x in comp_ctx_lengths_prefill.split(",")],
346-
default=[512],
347-
help='Define ccl list in csv format (e.g.,'
348-
'--comp-ctx-lengths 512,1024,2048).')
349-
parser.add_argument('--comp-ctx-lengths-decode',
350-
type=lambda comp_ctx_lengths_decode:
351-
[int(x) for x in comp_ctx_lengths_decode.split(",")],
352-
default=[2048],
353-
help='Define ccl list in csv format (e.g.,'
354-
'--comp-ctx-lengths 512,1024,2048).')
355343
parser.add_argument(
356344
"--comp-ctx-lengths-prefill",
357345
type=lambda comp_ctx_lengths_prefill: [int(x) for x in comp_ctx_lengths_prefill.split(",")],

0 commit comments

Comments
 (0)