SykoSLM commited on
Commit
b49fb5f
·
verified ·
1 Parent(s): 3dc2846

Update modeling_sykoslm.py

Browse files
Files changed (1) hide show
  1. modeling_sykoslm.py +2 -2
modeling_sykoslm.py CHANGED
@@ -5,10 +5,10 @@ from transformers import PreTrainedModel, PretrainedConfig
5
 
6
  class SykoSLMConfig(PretrainedConfig):
7
  model_type = "sykollm"
8
- def __init__(self, vocab_size=12000, d_model=512, n_layers=18, n_heads=4,
9
  num_memory_tokens=16, chunk_size=128, context_size=1024,
10
  overlap_size=16, code_overlap_size=64, abstract_head_hidden=256,
11
- abstract_head_layers=2, intermediate_size=2048, **kwargs):
12
  super().__init__(**kwargs)
13
  self.vocab_size = vocab_size
14
  self.d_model = d_model
 
5
 
6
  class SykoSLMConfig(PretrainedConfig):
7
  model_type = "sykollm"
8
+ def __init__(self, vocab_size=32000, d_model=768, n_layers=24, n_heads=6,
9
  num_memory_tokens=16, chunk_size=128, context_size=1024,
10
  overlap_size=16, code_overlap_size=64, abstract_head_hidden=256,
11
+ abstract_head_layers=2, intermediate_size=3072, **kwargs):
12
  super().__init__(**kwargs)
13
  self.vocab_size = vocab_size
14
  self.d_model = d_model