tyuan73 commited on
Commit
1ee0f1b
·
verified ·
1 Parent(s): b7ad068

Upload model

Browse files
Files changed (1) hide show
  1. config.json +7 -1
config.json CHANGED
@@ -1,4 +1,7 @@
1
  {
 
 
 
2
  "device": "cuda",
3
  "f_nums": [
4
  1,
@@ -17,6 +20,7 @@
17
  377,
18
  610
19
  ],
 
20
  "n_depth": 5,
21
  "n_embd": 738,
22
  "n_height": 3,
@@ -26,5 +30,7 @@
26
  "out_seq_size": 2048,
27
  "out_vocab_size": 350,
28
  "seq_size": 1024,
 
 
29
  "vocab_size": 50304
30
- }
 
1
  {
2
+ "architectures": [
3
+ "GoldenNet"
4
+ ],
5
  "device": "cuda",
6
  "f_nums": [
7
  1,
 
20
  377,
21
  610
22
  ],
23
+ "model_type": "golden_net",
24
  "n_depth": 5,
25
  "n_embd": 738,
26
  "n_height": 3,
 
30
  "out_seq_size": 2048,
31
  "out_vocab_size": 350,
32
  "seq_size": 1024,
33
+ "torch_dtype": "float32",
34
+ "transformers_version": "4.50.3",
35
  "vocab_size": 50304
36
+ }