xlstm-config: temporarily introduce new hidden_size parameter
Browse files- configuration_xlstm.py +1 -0
configuration_xlstm.py
CHANGED
@@ -32,6 +32,7 @@ class xLSTMConfig(PretrainedConfig):
|
|
32 |
self.vocab_size = vocab_size
|
33 |
self.embedding_dim = cfg.get("embedding_dim")
|
34 |
self.context_length = cfg.get("context_length")
|
|
|
35 |
|
36 |
def to_xlstm_config(self):
|
37 |
return from_dict(
|
|
|
32 |
self.vocab_size = vocab_size
|
33 |
self.embedding_dim = cfg.get("embedding_dim")
|
34 |
self.context_length = cfg.get("context_length")
|
35 |
+
self.hidden_size = cfg.get("embedding_dim")
|
36 |
|
37 |
def to_xlstm_config(self):
|
38 |
return from_dict(
|