suayptalha
commited on
Update configuration_minGRULM.py
Browse files
configuration_minGRULM.py
CHANGED
@@ -1,5 +1,3 @@
|
|
1 |
-
from transformers import PretrainedConfig
|
2 |
-
|
3 |
class MinGRULMConfig(PretrainedConfig):
|
4 |
model_type = "mingru"
|
5 |
|
@@ -9,8 +7,9 @@ class MinGRULMConfig(PretrainedConfig):
|
|
9 |
d_model=512,
|
10 |
ff_mult=4,
|
11 |
min_gru_expansion=1.5,
|
|
|
12 |
depth=12,
|
13 |
-
n_layer=12,
|
14 |
pad_vocab_size_multiple=8,
|
15 |
initializer_range=0.02,
|
16 |
**kwargs,
|
@@ -19,8 +18,9 @@ class MinGRULMConfig(PretrainedConfig):
|
|
19 |
self.d_model = d_model
|
20 |
self.ff_mult = ff_mult
|
21 |
self.min_gru_expansion = min_gru_expansion
|
|
|
22 |
self.depth = depth
|
23 |
-
self.n_layer = n_layer
|
24 |
self.pad_vocab_size_multiple = pad_vocab_size_multiple
|
25 |
self.initializer_range = initializer_range
|
26 |
-
super().__init__(**kwargs)
|
|
|
|
|
|
|
1 |
class MinGRULMConfig(PretrainedConfig):
|
2 |
model_type = "mingru"
|
3 |
|
|
|
7 |
d_model=512,
|
8 |
ff_mult=4,
|
9 |
min_gru_expansion=1.5,
|
10 |
+
expand=2.0, # Burada 'expand' parametresini ekledik
|
11 |
depth=12,
|
12 |
+
n_layer=12,
|
13 |
pad_vocab_size_multiple=8,
|
14 |
initializer_range=0.02,
|
15 |
**kwargs,
|
|
|
18 |
self.d_model = d_model
|
19 |
self.ff_mult = ff_mult
|
20 |
self.min_gru_expansion = min_gru_expansion
|
21 |
+
self.expand = expand # 'expand' parametresi eklendi
|
22 |
self.depth = depth
|
23 |
+
self.n_layer = n_layer
|
24 |
self.pad_vocab_size_multiple = pad_vocab_size_multiple
|
25 |
self.initializer_range = initializer_range
|
26 |
+
super().__init__(**kwargs)
|