KitsuVp commited on
Commit
bb2aaa5
·
verified ·
1 Parent(s): 36727ef

Update configuration_neollm.py

Browse files
Files changed (1) hide show
  1. configuration_neollm.py +1 -1
configuration_neollm.py CHANGED
@@ -575,7 +575,7 @@ class NeoLLMConfig(PretrainedConfig):
575
  use_laurel_lr=False,
576
  laurel_lr_rank=32,
577
  # ── Interleaved Head Attention (Duvvuri et al., 2026) ─────────────
578
- use_iha=False,
579
  iha_num_pseudo_heads=2, # P=2 → 2×2=4 patrones por head
580
  iha_local_global_pattern="LLLLG", # 4 locales + 1 global (paper §5.1)
581
  iha_sliding_window=None, # auto = N // (2*P^2) usando la longitud real del batch
 
575
  use_laurel_lr=False,
576
  laurel_lr_rank=32,
577
  # ── Interleaved Head Attention (Duvvuri et al., 2026) ─────────────
578
+ use_iha=True,
579
  iha_num_pseudo_heads=2, # P=2 → 2×2=4 patrones por head
580
  iha_local_global_pattern="LLLLG", # 4 locales + 1 global (paper §5.1)
581
  iha_sliding_window=None, # auto = N // (2*P^2) usando la longitud real del batch