Skip to content

Commit 3305f9a

Browse files
authored
Merge pull request #1 from eltociear/patch-1
chore: update utils.py
2 parents fa81707 + 66f167d commit 3305f9a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

chai_lab/model/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def get_qkv_indices_for_blocks(
3434
kv_block_size, device=kv_indices.device
3535
) # bl bl_kv -> kv
3636
# mask out positions where kv_indices gets wrapped
37-
# Rationale: the local attention block should allways process
37+
# Rationale: the local attention block should always process
3838
# local blocks (i.e. same rel-positional encodings for each block.)
3939
kv_mask = (kv_indices < sequence_length) & (kv_indices >= 0)
4040
# Use of % not .clamp is important for short sequences

0 commit comments

Comments
 (0)