PatrickHaller commited on
Commit
e6bb3d8
1 Parent(s): 4e40247

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +40 -0
config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "torch_dtype": "float32",
3
+ "architectures": [
4
+ "xLSTMForCausalLM"
5
+ ],
6
+ "transformers_version": "4.44.1",
7
+ "_xlstm_config": {
8
+ "num_blocks": 48,
9
+ "embedding_dim": 2048,
10
+ "mlstm_block": {
11
+ "mlstm": {
12
+ "num_heads": 4
13
+ }
14
+ },
15
+ "slstm_block": {
16
+ "slstm": {
17
+ "num_heads": 4
18
+ }
19
+ },
20
+ "slstm_at": [
21
+ 3,
22
+ 5,
23
+ 7,
24
+ 40,
25
+ 42,
26
+ 44
27
+ ],
28
+ "context_length": 512,
29
+ "vocab_size": 16000
30
+ },
31
+ "vocab_size": 16000,
32
+ "embedding_dim": 2048,
33
+ "context_length": 512,
34
+ "model_type": "xlstm",
35
+ "auto_map": {
36
+ "AutoConfig": "configuration_xlstm.xLSTMConfig",
37
+ "AutoModelForCausalLM": "modeling_xlstm.xLSTMForCausalLM",
38
+ "AutoModel": "modeling_xlstm.xLSTMModel"
39
+ }
40
+ }