wincode commited on
Commit
92d6984
·
verified ·
1 Parent(s): 0ac087f

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +28 -0
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "aetherstory",
3
+ "architectures": [
4
+ "AetherStoryModel"
5
+ ],
6
+ "library_name": "custom",
7
+ "language": "en",
8
+ "license": "mit",
9
+ "tags": [
10
+ "from-scratch",
11
+ "storytelling",
12
+ "cpu-trained",
13
+ "transformer"
14
+ ],
15
+ "vocab_size": 516,
16
+ "d_model": 128,
17
+ "n_layers": 4,
18
+ "n_heads": 4,
19
+ "ffn_dim": 512,
20
+ "max_seq_len": 64,
21
+ "dropout": 0.1,
22
+ "tie_embeddings": true,
23
+ "pad_token_id": 0,
24
+ "bos_token_id": 1,
25
+ "eos_token_id": 2,
26
+ "unk_token_id": 3,
27
+ "total_parameters": 863492
28
+ }