Refactor sliding window configuration to Transformers best practice (#21927)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -144,13 +144,10 @@ class Gemma2Attention(nn.Module):
|
||||
is_neox_style=True,
|
||||
)
|
||||
|
||||
# reference:
|
||||
# https://github.com/huggingface/transformers/blob/54be2d7ae87e873482b984cc956e165ca4dc0ba3/src/transformers/models/gemma2/modeling_gemma2.py#L312 # noqa
|
||||
layer_idx = extract_layer_index(prefix)
|
||||
use_sliding_window = (layer_idx % 2 == 0 and getattr(
|
||||
config, "interleaved_sliding_window", None) is not None)
|
||||
sliding_window = config.interleaved_sliding_window if \
|
||||
use_sliding_window else None
|
||||
is_sliding = config.layer_types[layer_idx] == "sliding_attention"
|
||||
sliding_window = config.sliding_window if is_sliding else None
|
||||
|
||||
self.attn = Attention(self.num_heads,
|
||||
self.head_dim,
|
||||
self.scaling,
|
||||
|
||||
Reference in New Issue
Block a user