Update configuration_minGRULM.py

configuration_minGRULM.py

@@ -9,20 +9,22 @@ class MinGRULMConfig(PretrainedConfig):
         d_model=512,
         ff_mult=4,
         min_gru_expansion=1.5,
-        expand=2.0,  # Burada 'expand' parametresini ekledik
+        expand=2.0,
         depth=12,
         n_layer=12,
         pad_vocab_size_multiple=8,
         initializer_range=0.02,
+        device=None,  # device parametresi eklendi
         **kwargs,
     ):
         self.vocab_size = vocab_size
         self.d_model = d_model
         self.ff_mult = ff_mult
         self.min_gru_expansion = min_gru_expansion
-        self.expand = expand  # 'expand' parametresi eklendi
+        self.expand = expand
         self.depth = depth
         self.n_layer = n_layer
         self.pad_vocab_size_multiple = pad_vocab_size_multiple
         self.initializer_range = initializer_range
+        self.device = device  # device parametresi
         super().__init__(**kwargs)