Skip to content

Commit 283d752

Browse files
committed
address feedback: renaming var
Signed-off-by: Yannick Schnider <[email protected]>
1 parent 07ec941 commit 283d752

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/offline_inference/long_context.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def round_up(t):
122122

123123

124124
tokens_to_generate = [
125-
args.max_model_len + 1 - round_up(plen) for plen in prompt_lens
125+
args.max_model_len + 1 - round_up(prompt_len) for prompt_len in prompt_lens
126126
]
127127

128128
sampling_params = [

0 commit comments

Comments
 (0)