xlr8harder commited on
Commit
410eeb2
·
verified ·
1 Parent(s): a35566a

Correct Talkie configuration default to 4k

Browse files
Files changed (1) hide show
  1. configuration_talkie.py +1 -1
configuration_talkie.py CHANGED
@@ -15,7 +15,7 @@ class TalkieConfig(PretrainedConfig):
15
  n_head: int = 40,
16
  n_embd: int = 5120,
17
  head_dim: int = 128,
18
- max_position_embeddings: int = 2048,
19
  rope_base: int = 1_000_000,
20
  rope_scaling: dict | None = None,
21
  rope_parameters: dict | None = None,
 
15
  n_head: int = 40,
16
  n_embd: int = 5120,
17
  head_dim: int = 128,
18
+ max_position_embeddings: int = 4096,
19
  rope_base: int = 1_000_000,
20
  rope_scaling: dict | None = None,
21
  rope_parameters: dict | None = None,