File size: 800 Bytes
2a6d57e 1746bec 2a6d57e 6ca2fe9 2a6d57e 1746bec 6ca2fe9 2a6d57e 1746bec 2a6d57e 6ca2fe9 1746bec 6ca2fe9 2a6d57e 1746bec 2a6d57e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from transformers import PretrainedConfig
class MinGRULMConfig(PretrainedConfig):
model_type = "mingru"
def __init__(
self,
vocab_size=50257,
d_model=512,
ff_mult=4,
min_gru_expansion=1.5,
depth=12,
n_layer=12, # Burada 'n_layer' parametresini ekledik
pad_vocab_size_multiple=8,
initializer_range=0.02,
**kwargs,
):
self.vocab_size = vocab_size
self.d_model = d_model
self.ff_mult = ff_mult
self.min_gru_expansion = min_gru_expansion
self.depth = depth
self.n_layer = n_layer # 'n_layer' parametresi eklendi
self.pad_vocab_size_multiple = pad_vocab_size_multiple
self.initializer_range = initializer_range
super().__init__(**kwargs) |