KitsuVp commited on
Commit
bdb5b28
Β·
verified Β·
1 Parent(s): bea8374

Update configuration_neollm.py

Browse files
Files changed (1) hide show
  1. configuration_neollm.py +3 -3
configuration_neollm.py CHANGED
@@ -572,7 +572,7 @@ class NeoLLMConfig(PretrainedConfig):
572
  mea_groupnorm_eps=1e-6,
573
  use_lucid_attention=False,
574
  lucid_attention_eps=1e-6,
575
- use_affine_scaled_attention=True,
576
  affine_momentum=0.9,
577
  use_xsa=True,
578
  xsa_eps=1e-6,
@@ -620,7 +620,7 @@ class NeoLLMConfig(PretrainedConfig):
620
  repo_start_layer=None,
621
  repo_d_p=None,
622
  use_repo_grape=True,
623
- use_repo_goat_prior=True,
624
  repo_goat_num_frequencies=3,
625
  repo_goat_sink_decay=4.0,
626
  # ── VersatileFFN (Nie et al., 2026) ───────────────────────────────
@@ -638,7 +638,7 @@ class NeoLLMConfig(PretrainedConfig):
638
  use_laurel_lr=False,
639
  laurel_lr_rank=32,
640
  # ── Interleaved Head Attention (Duvvuri et al., 2026) ─────────────
641
- use_iha=False,
642
  iha_num_pseudo_heads=2, # P=2 β†’ 2Γ—2=4 patrones por head
643
  iha_local_global_pattern="LLLLG", # 4 locales + 1 global (paper Β§5.1)
644
  iha_sliding_window=None, # auto = N // (2*P^2) usando la longitud real del batch
 
572
  mea_groupnorm_eps=1e-6,
573
  use_lucid_attention=False,
574
  lucid_attention_eps=1e-6,
575
+ use_affine_scaled_attention=False,
576
  affine_momentum=0.9,
577
  use_xsa=True,
578
  xsa_eps=1e-6,
 
620
  repo_start_layer=None,
621
  repo_d_p=None,
622
  use_repo_grape=True,
623
+ use_repo_goat_prior=False,
624
  repo_goat_num_frequencies=3,
625
  repo_goat_sink_decay=4.0,
626
  # ── VersatileFFN (Nie et al., 2026) ───────────────────────────────
 
638
  use_laurel_lr=False,
639
  laurel_lr_rank=32,
640
  # ── Interleaved Head Attention (Duvvuri et al., 2026) ─────────────
641
+ use_iha=True,
642
  iha_num_pseudo_heads=2, # P=2 β†’ 2Γ—2=4 patrones por head
643
  iha_local_global_pattern="LLLLG", # 4 locales + 1 global (paper Β§5.1)
644
  iha_sliding_window=None, # auto = N // (2*P^2) usando la longitud real del batch