stefan-it commited on
Commit
dbe6e99
1 Parent(s): 5980dab

xlstm-config: temporarily introduce new hidden_size parameter

Browse files
Files changed (1) hide show
  1. 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(