Jumpr commited on
Commit
f2906bf
·
verified ·
1 Parent(s): 9bbe923

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)
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b02bc8d7ec6d747303b8c350058ce137860de6bb131e0b066564cdf7e41e0f8c
3
  size 1772752
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47be34e4848987893e1a2496f517e13aead02c5eafbf14ec5c5927bf1883258a
3
  size 1772752