Jumpr commited on
Commit
4799474
·
verified ·
1 Parent(s): c7d303a

Upload model

Browse files
config.json CHANGED
@@ -21,5 +21,6 @@
21
  },
22
  "dtype": "float32",
23
  "model_type": "lightning_transformer",
 
24
  "transformers_version": "5.12.1"
25
  }
 
21
  },
22
  "dtype": "float32",
23
  "model_type": "lightning_transformer",
24
+ "num_hidden_layers": 4,
25
  "transformers_version": "5.12.1"
26
  }
configuration_lightningtransformer.py CHANGED
@@ -5,5 +5,6 @@ class LightningTransformerModelConfig(PretrainedConfig):
5
 
6
  def __init__(self, cfg=None, **kwargs):
7
  self.cfg = cfg
 
8
 
9
  super().__init__(**kwargs)
 
5
 
6
  def __init__(self, cfg=None, **kwargs):
7
  self.cfg = cfg
8
+ self.num_hidden_layers = 4
9
 
10
  super().__init__(**kwargs)