suayptalha commited on
Commit
6a0b190
·
verified ·
1 Parent(s): 6ca2b25

Update configuration_minGRULM.py

Browse files
Files changed (1) hide show
  1. configuration_minGRULM.py +2 -0
configuration_minGRULM.py CHANGED
@@ -14,6 +14,7 @@ class MinGRULMConfig(PretrainedConfig):
14
  n_layer=12,
15
  pad_vocab_size_multiple=8,
16
  initializer_range=0.02,
 
17
  **kwargs,
18
  ):
19
  self.vocab_size = vocab_size
@@ -23,6 +24,7 @@ class MinGRULMConfig(PretrainedConfig):
23
  self.expand = expand
24
  self.depth = depth
25
  self.n_layer = n_layer
 
26
  self.pad_vocab_size_multiple = pad_vocab_size_multiple
27
  self.initializer_range = initializer_range
28
  super().__init__(**kwargs)
 
14
  n_layer=12,
15
  pad_vocab_size_multiple=8,
16
  initializer_range=0.02,
17
+ hidden_size = 512,
18
  **kwargs,
19
  ):
20
  self.vocab_size = vocab_size
 
24
  self.expand = expand
25
  self.depth = depth
26
  self.n_layer = n_layer
27
+ self.hidden_size = hidden_size
28
  self.pad_vocab_size_multiple = pad_vocab_size_multiple
29
  self.initializer_range = initializer_range
30
  super().__init__(**kwargs)