Skip to content

Commit a5fdeca

Browse files
committed
latency script small fix
1 parent 50c55e0 commit a5fdeca

File tree

1 file changed

+1
-1
lines changed
  • users/zeyer/experiments/exp2023_02_16_chunked_attention/scripts

1 file changed

+1
-1
lines changed

users/zeyer/experiments/exp2023_02_16_chunked_attention/scripts/latency.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -402,7 +402,7 @@ def get_chunk_ends(deps: Deps, segment_name: str) -> List[Decimal]:
402402
print(bpe_labels)
403403
print(bpe_labels_s)
404404
chunk_idx = 0
405-
cur_chunk_end_pos = deps.chunk_left_padding + deps.chunk_size
405+
cur_chunk_end_pos = -deps.chunk_left_padding + deps.chunk_size
406406
cur_word = ""
407407
word_idx = 0
408408
res = []

0 commit comments

Comments
 (0)