We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents fa81707 + 66f167d commit 3305f9aCopy full SHA for 3305f9a
chai_lab/model/utils.py
@@ -34,7 +34,7 @@ def get_qkv_indices_for_blocks(
34
kv_block_size, device=kv_indices.device
35
) # bl bl_kv -> kv
36
# mask out positions where kv_indices gets wrapped
37
- # Rationale: the local attention block should allways process
+ # Rationale: the local attention block should always process
38
# local blocks (i.e. same rel-positional encodings for each block.)
39
kv_mask = (kv_indices < sequence_length) & (kv_indices >= 0)
40
# Use of % not .clamp is important for short sequences
0 commit comments