Update configuration_neollm.py
Browse files- 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=
|
| 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=
|
| 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=
|
| 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
|