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